MicroRNA classification and discovery for major depressive disorder diagnosis: Towards a robust and interpretable machine learning approach.

ABSTRACT

BACKGROUND: Major depressive disorder (MDD) is notably underdiagnosed and undertreated due to its complex nature and subjective diagnostic methods. Biomarker identification would help provide a clearer understanding of MDD aetiology. Although machine learning (ML) has been implemented in previous studies to study the alteration of microRNA (miRNA) levels in MDD cases, clinical translation has not been feasible due to the lack of interpretability (i.e. too many miRNAs for consideration) and stability. METHODS: This study applied logistic regression (LR) model to the blood miRNA expression profile to differentiate patients with MDD (n = 60) from healthy controls (HCs, n = 60). Embedded (L1-regularised logistic regression) feature selector was utilised to extract clinically relevant miRNAs, and optimized for clinical application. RESULTS: Patients with MDD could be differentiated from HCs with the area under the receiver operating characteristic curve (AUC) of 0.81 on testing data when all available miRNAs were considered (which served as a benchmark). Our LR model selected miRNAs up to 5 (known as LR-5 model) emerged as the best model because it achieved a moderate classification ability (AUC = 0.75), relatively high interpretability (feature number = 5) and stability (ϕ̂Z=0.55) compared to the benchmark. The top-ranking miRNAs identified by our model have demonstrated associations with MDD pathways involving cytokine signalling in the immune system, the reelin signalling pathway, programmed cell death and cellular responses to stress. CONCLUSION: The LR-5 model, which is optimised based on ML design factors, may lead to a robust and clinically usable MDD diagnostic tool.