Robust inference for risk heterogeneity under group imbalance

• URL: http://arxiv.org/abs/2606.00797v1
• Date: Sat, 30 May 2026 16:35:45 GMT
• Title: Robust inference for risk heterogeneity under group imbalance
• Abstract summary: In intensive care unit studies, mortality risk associated with specific admission diagnoses can vary across ethnic groups.<n>Existing approaches for detecting risk heterogeneity are often sensitive to baseline model misspecification and regularization bias.<n>We propose a robust framework for inferring risk heterogeneity between two populations using Neymanity estimators that are locally insensitive to nuisance parameter estimation error.
• Score: 8.713612612607408
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: Population-level heterogeneity is ubiquitous in biomedical data, where differences across demographic or clinical subgroups can substantially alter risk patterns. For example, in intensive care unit (ICU) studies, the mortality risk associated with specific admission diagnoses can vary across ethnic groups. Existing approaches for detecting risk heterogeneity are often sensitive to baseline model misspecification and regularization bias, both of which commonly arise in practice. In this paper, we propose a robust framework for inferring risk heterogeneity between two populations using Neyman orthogonality, which yields estimators that are locally insensitive to nuisance parameter estimation error. The proposed estimator is consistent and asymptotically normal, and simulation studies demonstrate that in finite samples our method substantially reduces bias and improves inferential stability compared with standard likelihood-based approaches. In an application to the eICU Collaborative Research Database, our method reveals clinically meaningful ethnicity-specific heterogeneity in admission diagnoses for in-hospital mortality that standard likelihood-based methods fail to detect.

Related papers

• Dementia Etiology Diagnosis via Collaborative Meta Knowledge Enhancement [81.27196033392559]
  We propose a Collaborative Meta Knowledge Enhancement (COME) framework for dementia etiology diagnosis.<n>COME injects multi-center acquisition semantics, source identifiers, and modality indicators as heterogeneity-aware embeddings into a unified Transformer architecture for scale-up training.<n>Our method achieves state-of-the-art in-domain performance with a mean macro-averaged AUC of 85.62% and a 4.29-point gain over the strongest baseline.
  arXiv  Detail & Related papers  (2026-07-24T04:26:47Z)
• Improving Generalizability of Hip Fracture Risk Prediction via Domain Adaptation Across Multiple Cohorts [15.756482189835566]
  We use a shared set of clinical and DXA-derived features across three large cohorts to evaluate the performance of three domain adaptation methods.<n>For a source cohort with males only, domain-adaptation methods consistently showed improved performance than the no-adaptation baseline.<n>Our outcome-free approaches enable the model selection under realistic deployment conditions and improve generalization of models in hip fracture risk prediction.
  arXiv  Detail & Related papers  (2026-02-20T03:37:05Z)
• Debiased maximum-likelihood estimators for hazard ratios under kernel-based machine-learning adjustment [0.0]
  We propose abandoning the baseline hazard and using kernel-based machine learning to explicitly model the change in the risk set with or without latent variables.<n> Numerical simulations confirm that the proposed method identifies the true hazard ratios with minimal bias.
  arXiv  Detail & Related papers  (2025-07-23T16:51:09Z)
• Unsupervised risk factor identification across cancer types and data modalities via explainable artificial intelligence [0.0]
  We present a novel method for unsupervised machine learning that directly optimize for survival heterogeneity across patient clusters.<n>Our approach represents novel methodology for training any neural network architecture on any data modality to identify prognostically distinct patient groups.<n>This pan-cancer, model-agnostic approach represents a valuable advancement in clinical risk stratification.
  arXiv  Detail & Related papers  (2025-06-15T19:11:10Z)
• Adaptable Cardiovascular Disease Risk Prediction from Heterogeneous Data using Large Language Models [70.64969663547703]
  AdaCVD is an adaptable CVD risk prediction framework built on large language models extensively fine-tuned on over half a million participants from the UK Biobank.<n>It addresses key clinical challenges across three dimensions: it flexibly incorporates comprehensive yet variable patient information; it seamlessly integrates both structured data and unstructured text; and it rapidly adapts to new patient populations using minimal additional data.
  arXiv  Detail & Related papers  (2025-05-30T14:42:02Z)
• Assessing Surrogate Heterogeneity in Real World Data Using Meta-Learners [0.0]
  We propose a framework to assess surrogate heterogeneity in real-world, non-randomized, data.<n>Our approach allows us to quantify heterogeneity in surrogate strength with respect to patient characteristics.<n>We use our framework to identify individuals for whom the surrogate is a valid replacement of the primary outcome.
  arXiv  Detail & Related papers  (2025-04-21T18:54:48Z)
• Seeing Unseen: Discover Novel Biomedical Concepts via Geometry-Constrained Probabilistic Modeling [53.7117640028211]
  We present a geometry-constrained probabilistic modeling treatment to resolve the identified issues. We incorporate a suite of critical geometric properties to impose proper constraints on the layout of constructed embedding space. A spectral graph-theoretic method is devised to estimate the number of potential novel classes.
  arXiv  Detail & Related papers  (2024-03-02T00:56:05Z)
• Targeting relative risk heterogeneity with causal forests [1.2255617580795168]
  We propose and implement a methodology for modifying causal forests to target relative risk.<n>We present results from simulated data that suggest relative risk causal forests can capture otherwise undetected sources of heterogeneity.<n>We implement our method on real-world trial data to explore HTEs for liraglutide in patients with type 2 diabetes.
  arXiv  Detail & Related papers  (2023-09-26T16:57:46Z)
• Benchmarking common uncertainty estimation methods with histopathological images under domain shift and label noise [62.997667081978825]
  In high-risk environments, deep learning models need to be able to judge their uncertainty and reject inputs when there is a significant chance of misclassification. We conduct a rigorous evaluation of the most commonly used uncertainty and robustness methods for the classification of Whole Slide Images. We observe that ensembles of methods generally lead to better uncertainty estimates as well as an increased robustness towards domain shifts and label noise.
  arXiv  Detail & Related papers  (2023-01-03T11:34:36Z)
• Hierarchical Semi-Supervised Contrastive Learning for Contamination-Resistant Anomaly Detection [81.07346419422605]
  Anomaly detection aims at identifying deviant samples from the normal data distribution. Contrastive learning has provided a successful way to sample representation that enables effective discrimination on anomalies. We propose a novel hierarchical semi-supervised contrastive learning framework, for contamination-resistant anomaly detection.
  arXiv  Detail & Related papers  (2022-07-24T18:49:26Z)
• Mitigating multiple descents: A model-agnostic framework for risk monotonization [84.6382406922369]
  We develop a general framework for risk monotonization based on cross-validation. We propose two data-driven methodologies, namely zero- and one-step, that are akin to bagging and boosting.
  arXiv  Detail & Related papers  (2022-05-25T17:41:40Z)
• Survival Cluster Analysis [93.50540270973927]
  There is an unmet need in survival analysis for identifying subpopulations with distinct risk profiles. An approach that addresses this need is likely to improve characterization of individual outcomes.
  arXiv  Detail & Related papers  (2020-02-29T22:41:21Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.