Progressive Neural Networks for Continuous Classification of Retinal Optical Coherence Tomography Images

Di Wang,Kun Huang,Qiang Chen
DOI: https://doi.org/10.1109/cbd63341.2023.00036
2023-01-01
Abstract:Deep learning models have demonstrated a great effectiveness on the classification of retinal lesions in optical coherence tomography images. However, the performance of these models deteriorates significantly when classifying images from different devices compared to the training data, which greatly limits their practicality. To address the issue of poor performance of the model in real-world applications, we propose a method of multi-granularity feature space constraint by studying samples from different devices as domain incremental tasks, to maintain the stability and plasticity of the model, thereby making the model having good generalization performance. Experimental results demonstrate that the proposed method effectively maintains performance on both new and old datasets. Compared to some state-of-the-art baselines focusing on natural images, the proposed method in this paper achieves higher average accuracy and lower forgetting rate on publicly available medical image datasets.
What problem does this paper attempt to address?