# MOSIC: Model-Agnostic Optimal Subgroup Identification with Multi-Constraint for Improved Reliability Source: [MOSIC: Model-Agnostic Optimal Subgroup Identification with Multi-Constraint for Improved Reliability]() Collection index: [llms.txt]() Data updated: 2026-06-16T05:00:05.515Z Authors: Wenxin Chen, Weishen Pan, Kyra Gan, Fei Wang Published: 2025-04-29T16:25:23.000Z ## Abstract Current subgroup identification methods typically follow a two-step approach: first estimate conditional average treatment effects and then apply thresholding or rule-based procedures to define subgroups. While intuitive, this decoupled approach fails to incorporate key constraints essential for real-world clinical decision-making, such as subgroup size and propensity overlap. These constraints operate on fundamentally different axes than CATE estimation and are not naturally accommodated within existing frameworks, thereby limiting the practical applicability of these methods. We propose a unified optimization framework that directly solves the primal constrained optimization problem to identify optimal subgroups. Our key innovation is a reformulation of the constrained primal problem as an unconstrained differentiable min-max objective, solved via a gradient descent-ascent algorithm. We theoretically establish that our solution converges to a feasible and locally optimal solution. Unlike threshold-based CATE methods that apply constraints as post-hoc filters, our approach enforces them directly during optimization. The framework is model-agnostic, compatible with a wide range of CATE estimators, and extensible to additional constraints like cost limits or fairness criteria. Extensive experiments on synthetic and real-world datasets demonstrate its effectiveness in identifying high-benefit subgroups while maintaining better satisfaction of constraints. ## Recorded benchmark results A research result applies to its stated dataset, split, metric and evaluation configuration. Compare like-for-like settings, preserve the metric's unit and direction, and distinguish paper-reported results from independent reproduction. The best result recorded here is not a claim of exhaustive, current global coverage. - [Subgroup Identification on eICU ventilation-free-days outcome](): Subgroup Identification; eICU ventilation-free-days outcome; ESS: 531.92. - [Optimal Subgroup Identification on eICU ventilation-free days within 7 days (test)](): Optimal Subgroup Identification; eICU ventilation-free days within 7 days (test); Subgroup ATE: 0.05. ## Original sources and implementation [Original paper]() [PDF]()