Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This study introduces NITROGEN, an imputation-free transformer model designed to enhance Alzheimer's disease prediction by effectively managing incomplete and heterogeneous clinical data. By employing masked and intersample attention, NITROGEN captures both within-patient feature dependencies and between-patient relationships, leading to improved calibration and uncertainty quantification compared to traditional tree-based methods. The model was validated across multiple cohorts, revealing critical features for classification and demonstrating that robust performance in clinical settings requires a focus on calibration and interpretability rather than accuracy alone.
NITROGEN achieves superior uncertainty quantification and calibration in Alzheimer's predictions without relying on imputation, challenging conventional methods that often introduce bias.
Accurate diagnostic classification and disease-severity prediction for Alzheimer's disease are hampered by the incompleteness and heterogeneity of real-world clinical data. Left unaddressed, these barriers prevent reliable disease modelling and hinder effective clinical evaluation. Conventional imputation strategies introduce systematic bias, distort inter-feature relationships, and yield overconfident predictions, limitations especially consequential in diagnostic settings. Here, we propose NITROGEN, an imputation-free transformer that jointly models within-patient feature dependencies and between-patient relational structure through masked and intersample attention, enabling robust multimodal learning directly from partially observed records. We trained NITROGEN on ADNI (N=7858 scans), and evaluated it on two independent cohorts: OASIS-3 (N=2675 scans) and AIBL (N=1286 scans). Across cohorts and diagnostic and cognitive score prediction tasks, NITROGEN showed robust calibration and uncertainty quantification advantages over tree-based ensemble methods, while maintaining competitive discriminative performance. Cross-cohort and cross-method analyses identified cortical thickness in the temporal pole, age, and APOE genotype as important, though not individually sufficient, features for AD classification. We further introduced a modality-aware uncertainty adjustment that augments predictive uncertainty proportionally to the importance of absent modalities, enabling calibrated confidence when diagnostic information is unavailable. Together, our results show that imputation-free attention learning preserved meaningful discrimination under cohort shift, revealing expected degradation on more distributionally different cohorts, and demonstrate that evaluating models along calibration, interpretability, and cross-cohort reliability, not accuracy alone, is essential for clinical deployment.