Previous Article in Journal
HDPF: Hierarchical Dual-Perspective Collaborative Modeling for Multimodal Image Fusion
Previous Article in Special Issue
Explainable Artificial Intelligence (XAI) for EEG Analysis: A Survey on Recent Trends and Advancements
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
This is an early access version, the complete PDF, HTML, and XML versions will be available soon.
Article

Interpretable Subgroup Discovery with Abstention in Small, Heterogeneous Clinical Trials: A Retrospective Multi-Dataset Study

1
NetraMark Corp, Toronto, ON M6P 3S9, Canada
2
Department of Pathology and Molecular Medicine, Queen’s University, Kingston, ON K7L 3N6, Canada
3
Tanenbaum Centre for Pharmacogenetics, Centre for Addiction and Mental Health, Toronto, ON M5T 1R8, Canada
4
Arthur C. Clarke Center for Human Imagination, School of Physical Sciences, University of California, San Diego, La Jolla, CA 92065, USA
5
Department of Psychiatry and Behavioral Sciences, Leonard M. Miller School of Medicine, University of Miami, Coral Gables, FL 33136, USA
6
Department of Biomedical, Metabolic, and Neural Sciences, University of Modena and Reggio Emilia, 41125 Modena, Italy
*
Author to whom correspondence should be addressed.
AI 2026, 7(9), 349; https://doi.org/10.3390/ai7090349
Submission received: 22 July 2026 / Revised: 27 August 2026 / Accepted: 1 September 2026 / Published: 4 September 2026

Abstract

Small, heterogeneous clinical datasets pose a challenge for whole-cohort prediction because clinically meaningful treatment or response patterns may be diluted across biologically diverse patients. We describe and evaluate NetraAI, an interpretable dynamical-systems framework for selective subgroup discovery that uses finite-iteration contraction-inspired dynamics and long-range memory (LRM) to identify stable, outcome-linked Model-Derived Subgroups (MDS). This system can abstain by assigning No Call when a stable subgroup assignment is not supported. A large language model (LLM) Strategist is outlined only as a possible future extension; it is not evaluated here and contributes nothing to the results reported. Foundation and language models asked to perform subgroup discovery directly did not recover the structure the specialized discovery step recovered. We demonstrate this framework across three retrospective clinical trial datasets: Clinical Antipsychotic Trials of Intervention Effectiveness (CATIE) schizophrenia (olanzapine vs. perphenazine comparative treatment-preference benchmark), Canadian Biomarker Integration Network in Depression (CAN-BIND) depression (escitalopram response), and Comprehensive Molecular Characterization of Advanced Pancreatic Ductal Adenocarcinoma for Better Treatment Selection (COMPASS) pancreatic cancer (GnP vs. FOLFIRINOX observational regimen-associated response). The benchmark is not a contest between NetraAI and competing predictors: the same eight downstream methods are evaluated with and without what NetraAI discovered. Given the full feature sets and their own selection procedures, those methods were at or near chance on all three datasets, and blind de novo searches by an independent interaction model and by a pretrained tabular foundation model did not recover an equivalent signature or subpopulation. In internal downstream evaluation, given the discovered variables alone—the same patients, the same classifiers, the full cohort and no abstention of any kind—every one of the eight methods improved on every dataset, 24 of 24 method-dataset comparisons, moving from a raw-feature range of 0.46–0.62 AUC to 0.54–0.78. Restricting further to the subpopulation in which those variables hold improved all eight methods again in CAN-BIND and in COMPASS, 16 of 16 comparisons, reaching 0.66–0.83 and 0.96–1.00, respectively; in CATIE, where the called subgroups are the least outcome-homogeneous of the three, it improved only one of eight. Taking the framework as a whole, 23 of 24 method-dataset combinations improved over the raw-feature baseline. NetraAI abstains on patients without stable subgroup structure, calling 27.7% to 40.4% of each cohort. The contribution demonstrated is therefore subgroup discovery rather than downstream prediction, and its beneficiaries are the conventional methods themselves. In COMPASS, NetraAI identified a three-SNV signature associated with regimen-linked response ranking among called patients; because the cohort was observational and the permutation test was conditional on the selected signature, this finding is exploratory. The two mechanisms are complementary rather than competing: variable discovery establishes which features carry the structure, and abstention establishes in which patients it holds. Neither mechanism replaces conventional modeling. Variable discovery improved every method on every dataset, the pretrained tabular foundation model included; identifying the population in which those variables hold conferred further benefit in two of the three datasets and not in the third. These findings support NetraAI as an exploratory system for generating compact, inspectable subgroup hypotheses that may inform future enrichment strategies after external validation, and indicate that its value lies in what it contributes to other methods rather than in competing with them.
Keywords: artificial intelligence; interpretable machine learning; subgroup discovery; selective prediction; clinical trials; heterogeneous treatment effects; enrichment; foundation models; feature selection; abstention artificial intelligence; interpretable machine learning; subgroup discovery; selective prediction; clinical trials; heterogeneous treatment effects; enrichment; foundation models; feature selection; abstention

Share and Cite

MDPI and ACS Style

Geraci, J.; Qorri, B.; Cumbaa, C.; Tsay, M.; Marrella, C.A.; Zappulla, S.; Leonczyk, P.; Gogacz, A.; Pani, L. Interpretable Subgroup Discovery with Abstention in Small, Heterogeneous Clinical Trials: A Retrospective Multi-Dataset Study. AI 2026, 7, 349. https://doi.org/10.3390/ai7090349

AMA Style

Geraci J, Qorri B, Cumbaa C, Tsay M, Marrella CA, Zappulla S, Leonczyk P, Gogacz A, Pani L. Interpretable Subgroup Discovery with Abstention in Small, Heterogeneous Clinical Trials: A Retrospective Multi-Dataset Study. AI. 2026; 7(9):349. https://doi.org/10.3390/ai7090349

Chicago/Turabian Style

Geraci, Joseph, Bessi Qorri, Christian Cumbaa, Mike Tsay, Christopher Alexander Marrella, Seb Zappulla, Paul Leonczyk, Adam Gogacz, and Luca Pani. 2026. "Interpretable Subgroup Discovery with Abstention in Small, Heterogeneous Clinical Trials: A Retrospective Multi-Dataset Study" AI 7, no. 9: 349. https://doi.org/10.3390/ai7090349

APA Style

Geraci, J., Qorri, B., Cumbaa, C., Tsay, M., Marrella, C. A., Zappulla, S., Leonczyk, P., Gogacz, A., & Pani, L. (2026). Interpretable Subgroup Discovery with Abstention in Small, Heterogeneous Clinical Trials: A Retrospective Multi-Dataset Study. AI, 7(9), 349. https://doi.org/10.3390/ai7090349

Article Metrics

Back to TopTop