Many-MobileNet: Multi-Model Augmentation for Robust Retinal Disease Classification
2024-12-03Code Available2· sign in to hype
Hao Wang, Wenhui Zhu, Xuanzhao Dong, Yanxi Chen, Xin Li, Peijie Qiu, Xiwen Chen, Vamsi Krishna Vasa, Yujian Xiong, Oana M. Dumitrascu, Abolfazl Razi, Yalin Wang
Code Available — Be the first to reproduce this paper.
ReproduceCode
- github.com/Retinal-Research/Beyond-MobileNetOfficialpytorch★ 109
Abstract
In this work, we propose Many-MobileNet, an efficient model fusion strategy for retinal disease classification using lightweight CNN architecture. Our method addresses key challenges such as overfitting and limited dataset variability by training multiple models with distinct data augmentation strategies and different model complexities. Through this fusion technique, we achieved robust generalization in data-scarce domains while balancing computational efficiency with feature extraction capabilities.