arXiv AI By Seonghwi Kim, Sung Ho Jo, Minwoo Chae

Distributionally Robust Survival Models under Subpopulation Shift and Outlier Contamination

Read the original on arXiv AI →

The paper introduces a distributionally robust framework for survival analysis that simultaneously tackles latent subpopulation shift and outlier contamination. It employs an outer minimization to refine the nominal distribution by down-weighting contaminated samples and an inner maximization to target the most challenging subpopulation, directly handling non-decomposable survival losses such as the Cox partial log-likelihood. An alternating gradient-based algorithm, guided by KKT conditions, is developed, and experiments on simulated data and two benchmarks show improved worst-group performance and stable training under contamination.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

Hugging Face Trending Papers
Aug 13

Wasserstein Filtering: A Sample Selection Method for Robust Distribution Learning

Given a dataset where a portion of the samples are contaminated, our goal is to recover the underlying clean population distribution. To this end, we propose Wasserstein Filtering (WF), a novel sample selection framework that discards a fraction of suspicious samples and estimates the target distribution using the empirical measure of the remaining data.

arXiv Machine Learning
Jun 26

Learning from a Biased Sample

arXiv:2209. 01754v5 Announce Type: replace-cross Abstract: The empirical risk minimization approach to data-driven decision making requires access to training data drawn under the same conditions as those that will be faced when the decision rule is deployed.

By Roshni Sahoo, Lihua Lei, Stefan Wager