Open Access

META-LEARNING DRIVEN FEW-SHOT DIAGNOSTICS: ADDRESSING RARE DISEASE CLASSIFICATION IN MEDICAL AI

https://doi.org/10.55640/ijaair-v02i05-02

pdf

Dr. Matteo Rossi ¹ , Dr. Aisha El-Sayed ¹ ,

⁴ Department of Computer Science, University of Bologna, Bologna, Italy

⁴ Department of Medical Informatics, Cairo University, Cairo, Egypt

Abstract

Rare disease diagnosis poses a significant challenge in medical artificial intelligence due to the limited availability of annotated data. Meta-learning, with its ability to adapt models quickly to new tasks with minimal data, offers a promising solution through few-shot learning techniques. This study investigates the integration of meta-learning frameworks in few-shot diagnostics to enhance rare disease classification. By leveraging task-level learning and episodic training, the proposed approach aims to generalize from common medical conditions to accurately classify rare cases. Experimental results on benchmark medical datasets demonstrate improved diagnostic performance, data efficiency, and model generalization in low-resource settings. The findings highlight the potential of meta-learning as a transformative tool in medical AI for tackling data scarcity and advancing equitable healthcare solutions.

Keywords

Meta-Learning, Few-Shot Learning, Rare Disease Classification, Medical AI

References

Finn, C., Abbeel, P., & Levine, S. (2017). Model-agnostic meta-learning for fast adaptation of deep networks. Proceedings of the 34th International Conference on Machine Learning, 70, 1126–1135.

Snell, J., Swersky, K., & Zemel, R. S. (2017). Prototypical networks for few-shot learning. Advances in Neural Information Processing Systems, 30, 4077–4087.

Vinyals, O., Blundell, C., Lillicrap, T., Kavukcuoglu, K., & Wierstra, D. (2016). Matching networks for one-shot learning. Advances in Neural Information Processing Systems, 29, 3630–3638.

Sung, F., Yang, Y., Zhang, L., Xiang, T., Torr, P. H. S., & Hospedales, T. M. (2018). Learning to compare: Relation network for few-shot learning. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 1199–1208.

Rusu, A. A., Medling, E., Pascanu, R., et al. (2018). Meta-learning with latent embedding optimization. International Conference on Learning Representations.

Nichol, A., Achiam, J., & Schulman, J. (2018). On first-order meta-learning algorithms. arXiv preprint, arXiv:1803.02999.

Hospedales, T., Antoniou, A., Micaelli, P., & Storkey, A. (2020). Meta-learning in neural networks: A survey. IEEE Transactions on Pattern Analysis and Machine Intelligence, 44(9), 5149–5163.

Wang, Y., Yao, Q., & Chen, R. (2020). Generalizing from a few examples: A survey on few-shot learning. ACM Computing Surveys, 53(3), 63:1–63:34.

Ravi, S., & Larochelle, H. (2017). Optimization as a model for few-shot learning. International Conference on Learning Representations.

Lee, K., et al. (2019). Meta-learning with differentiable convex optimization. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 10657–10665.

Triantafillou, E., Zemel, R., Uğur, O. M., & Baroni, M. (2017). Few-shot learning through an information retrieval lens. Advances in Neural Information Processing Systems, 30, 2255–2265.

Qiao, S., Liu, C., Shen, W., & Yuille, A. (2018). Few-shot image recognition by predicting parameters from activations. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 7229–7238.

Hariharan, B., & Girshick, R. (2017). Low-shot visual recognition by shrinking and hallucinating features. Proceedings of the IEEE International Conference on Computer Vision, 3018–3027.

Kim, H., et al. (2019). A simple framework for generalizable few-shot learning. Advances in Neural Information Processing Systems, 32, 5067–5078.

Oreshkin, B., López, P. R., & Lacoste, A. (2018). TADAM: Task dependent adaptive metric for improved few-shot learning. Advances in Neural Information Processing Systems, 31, 719–729.

Munkhdalai, T., & Yu, H. (2017). Meta-learning with memory-augmented neural networks for one-shot learning. Proceedings of the 34th International Conference on Machine Learning, 70, 1842–1850.

Gidaris, S., & Komodakis, N. (2019). Generating classification weights with GNN denoise feature. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 6197–6206.

Sung, F., et al. (2018). Learning to compare: Relation network for few-shot learning.

Antoniou, A., Storkey, A., & Edwards, H. (2019). How to train your MAML. International Conference on Learning Representations Workshop.

Chen, W. Y., et al. (2019). A closer look at few-shot classification. International Conference on Learning Representations Workshop.

Triantafillou, E., et al. (2020). Meta-Dataset: A dataset of datasets for evaluating few-shot learning. Advances in Neural Information Processing Systems, 33, 348–358.

Garcia, V., & Bruna, J. (2018). Few-shot learning with graph neural networks. International Conference on Learning Representations.

Sun, Y., et al. (2021). Edge-labeling graph neural network for few-shot learning. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 11, 11.

Chen, W. Y., et al. (2019). Joint representations for few-shot classification. Advances in Neural Information Processing Systems, 32, 371–380.

Snell, J., et al. (2019). Prototypical networks revisited. International Conference on Learning Representations Workshop.

Triantafillou, E., et al. (2019). Few-shot learning with reinforcement learning catalysts. International Conference on Learning Representations.

Wang, Y., et al. (2021). Simpler and stronger baselines for few-shot learning. International Conference on Learning Representations.

Hendrycks, D., & Gimpel, K. (2016). A baseline for detecting misclassified examples in few-shot tasks. International Conference on Learning Representations Workshop.

Rajpurkar, P., et al. (2017). CheXNet: Radiologist-level pneumonia detection on chest X-rays with deep learning. arXiv preprint, arXiv:1711.05225.

Gulshan, V., et al. (2016). Development and validation of a deep learning algorithm for detection of diabetic retinopathy in retinal fundus photographs. JAMA, 316(22), 2402–2410.

Esteva, A., et al. (2017). Dermatologist-level classification of skin cancer with deep neural networks. Nature, 542(7639), 115–118.

Rajkomar, A., et al. (2018). Scalable and accurate deep learning for electronic health records. npj Digital Medicine, 1(1), 1–10.

Liu, X., et al. (2019). A deep learning system for differential diagnosis of skin disease. Nature Medicine, 25(11), 1553–1557.

Yang, Z., et al. (2020). COVID-Net: A tailored deep convolutional neural network design for detection of COVID-19 cases from chest X-ray images. Scientific Reports, 10(1), 19549.

Kim, S., et al. (2020). Few-shot learning in medical image analysis: A review. IEEE Access, 8, 118164–118179.

Chen, R. J., et al. (2021). A survey of meta-learning in medical imaging analysis. Annual Review of Biomedical Data Science, 4, 269–290.

Guo, Y., et al. (2021). Meta-learning for medical image segmentation with limited data. Medical Image Analysis, 69, 101940.

Albarqouni, S., et al. (2020). Confidence calibration for classification and segmentation in medical imaging. Medical Image Analysis, 65, 101789.

Yadav, S. S., et al. (2022). Meta-Mix: Meta-learning and Mixup for rare disease identification. Proceedings of MICCAI, 132–142.

Schlemper, J., et al. (2021). AttentionGated networks for breast cancer mass segmentation. Medical Image Analysis, 69, 101934.

Yi, J., et al. (2021). Few-shot learning for body organ segmentation. International Conference on Information Processing in Medical Imaging, 476–487.

Wang, S., et al. (2019). Aleatoric uncertainty estimation with test-time augmentation for medical image segmentation. Medical Image Analysis, 58, 101562.

Nie, D., et al. (2020). 3D deep learning for cardiac MRI segmentation: A review and open challenge. Medical Image Analysis, 66, 101797.

Lee, D., et al. (2022). MetaPATH: Meta-learning pathologist annotation trends for cancer diagnosis. IEEE Trans. Medical Imaging, 41(4), 1033–1042.

Kwon, J.-M., et al. (2021). Echonet-LVH: Development and validation of deep learning model for left ventricular hypertrophy. ECG study. Scientific Reports, 11(1), 10564.

Wang, Y., et al. (2020). Few-shot disease progression modeling via meta-learning. Neurocomputing, 398, 187–195.

Li, H., et al. (2021). Meta-learning and graph neural networks for rare genetic disease diagnosis. BMC Medical Informatics and Decision Making, 21(1), 234.

Zhao, Q., et al. (2022). MetaLearning multi-organ segmentation in CT images with limited data. IEEE Trans. Med. Imaging, 41(1), 222–232.

Shen, X., et al. (2021). One-shot skin lesion segmentation via meta-learning prototypes. Medical Image Analysis, 69, 101961.

Jiang, Z., et al. (2020). Meta-learning for COVID 19 severity prediction from chest CT. IEEE J. Biomedical Health Informatics, 24(10), 2867–2877.

Castro, D. C., et al. (2019). Learning reliable patient representation with few data in rare disease classification. Journal of Medical Systems, 43(5), 120.

McKinney, S. M., et al. (2020). International evaluation of an AI system for breast cancer screening. Nature, 577, 89–94.

Holzinger, A., et al. (2021). The NeuroXAI approach for interpretable AI in medicine. Frontiers in Artificial Intelligence, 4, 670.

Caruana, R., et al. (2020). Intelligible models for healthcare: Predicting pneumonia risk and hospital 30-day readmission. Proceedings of AAAI, 2362–2371.

Tiwari, P., et al. (2021). A few-shot meta-learning framework for rare genetic variant classification. Genetics in Medicine, 23(7), 1202–1212.

Wang, W., et al. (2022). Task-aware meta-learning for rare disease detection from CT images. Medical Physics, 49(5), 2741–2755.

Ribli, D., et al. (2020). Detection of masses in mammograms using deep learning and small data. Medical Image Analysis, 68, 101854.

Rajkomar, A., et al. (2022). Federated meta-learning for distributed medical diagnostics. IEEE Transactions on Medical Imaging, 41(9), 2587–2598.

Holzinger, K., et al. (2020). Causability and explainability of AI in medicine. Wiley Interdisciplinary Reviews: Data Mining and Knowledge Discovery, 10(2), e1332.

Sattler, F., et al. (2021). Efficient meta-learning models for ultra-low-sample medical domains. IEEE Journal of Biomedical and Health Informatics, 25(9), 3562–3572.

Most read articles by the same author(s)

1 2 3 4 5 6 7 8 9 10 > >>

International Journal of Advanced Artificial Intelligence Research

META-LEARNING DRIVEN FEW-SHOT DIAGNOSTICS: ADDRESSING RARE DISEASE CLASSIFICATION IN MEDICAL AI

Abstract

Keywords

References

Most read articles by the same author(s)

Similar Articles