A mixture of experts approach for enhanced diabetes detection
摘要
Accurate prediction of Type 2 and Gestational Diabetes is critical for timely intervention, personalized treatment, and effective disease management. The main purpose of this study is to develop a robust diabetes prediction framework capable of handling heterogeneous and imbalanced clinical data while improving minority-class detection. An enhanced Mixture of Experts (MoE) framework for diabetes prediction is proposed to improve classification performance under class imbalance conditions. The model employs a sparse gating mechanism that activates only the top-K most relevant experts for each sample, enabling efficient expert specialization and better generalization across diverse patient subgroups. To address expert workload imbalance, an innovative auxiliary loss based on the Gini index is introduced, which promotes balanced expert utilization with lower computational complexity than traditional coefficient-of-variation-based losses. Furthermore, an asymmetric focal loss is incorporated to mitigate class imbalance by reducing the dominance of negative samples during training and improving recall for diabetic cases. Experimental evaluations on two publicly available datasets for Type 2 and Gestational Diabetes show that the proposed MoE framework consistently outperforms conventional machine learning methods in terms of F1-score, recall, and precision–recall balance. These results show that the proposed framework improves diabetes prediction performance and provides a reliable tool for clinical decision support in imbalanced medical datasets.