{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/multi-disease-detection-in-retinal-imaging","title":"Multi-Disease Detection in Retinal Imaging based on Ensembling Heterogeneous Deep Learning Models","arxiv_id":"2103.14660","date":"2021-03-26","proceeding":null,"authors":["Dominik Müller","Iñaki Soto-Rey","Frank Kramer"],"abstract":"Preventable or undiagnosed visual impairment and blindness affect billion of people worldwide. Automated multi-disease detection models offer great potential to address this problem via clinical decision support in diagnosis. In this work, we proposed an innovative multi-disease detection pipeline for retinal imaging which utilizes ensemble learning to combine the predictive capabilities of several heterogeneous deep convolutional neural network models. Our pipeline includes state-of-the-art strategies like transfer learning, class weighting, real-time image augmentation and Focal loss utilization. Furthermore, we integrated ensemble learning techniques like heterogeneous deep learning models, bagging via 5-fold cross-validation and stacked logistic regression models. Through internal and external evaluation, we were able to validate and demonstrate high accuracy and reliability of our pipeline, as well as the comparability with other state-of-the-art pipelines for retinal disease prediction.","url_abs":"https://arxiv.org/abs/2103.14660v1","url_pdf":"https://arxiv.org/pdf/2103.14660v1.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"multi-disease-detection-in-retinal-imaging","repo_url":"https://github.com/frankkramer-lab/riadd.aucmedi","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"tf","reach":null},{"paper_slug":"multi-disease-detection-in-retinal-imaging","repo_url":"https://github.com/ashokkumar-kola/FYP","is_official":0,"mentioned_in_paper":0,"mentioned_in_github":1,"framework":"tf","reach":null}],"tasks":[{"task_slug":"disease-prediction","task_name":"Disease Prediction"},{"task_slug":"ensemble-learning","task_name":"Ensemble Learning"},{"task_slug":"image-augmentation","task_name":"Image Augmentation"},{"task_slug":"transfer-learning","task_name":"Transfer Learning"}],"methods":[{"method_slug":"focal-loss","method_name":"Focal Loss"},{"method_slug":"logistic-regression","method_name":"Logistic Regression"}],"datasets_introduced":[],"methods_introduced":[],"results":[{"leaderboard":"/sota/transfer-learning-on-retinal-fundus","task":"Transfer Learning","dataset":"Retinal Fundus MultiDisease Image Dataset (RFMiD)","model":"riadd.aucmedi","rank_in_archive_order":1,"of":1,"metrics":{"AUROC":"0.95"},"uses_additional_data":false}],"syntology":{"syntology_url":null,"atlas_url":null,"mcp":null,"developers":"https://syntology.ai/developers"},"arxiv_metadata":null,"syntology_extracted_results":null}