IEEE Access (Jan 2024)
A Novel Artificial-Intelligence-Based Approach for Automatic Assessment of Retinal Disease Images Using Multi-View Deep-Broad Learning Network
Abstract
Retinal disease detection and diagnosis relying solely on artificial retinal diseases will put great pressure on doctors and increase the rate of misdiagnosis. Therefore, the development of computer vision technology has brought the possibility for ophthalmologists to use computer-aided diagnosis. In recent years, most models for retinal disease recognition have been based on deep learning, which has the disadvantage of requiring large amounts of data and training time. It is also partly based on broad learning and its disadvantages are that feature extraction ability is limited and poor scalability. To overcome these limitations, we propose a novel artificial intelligence-based approach for the automatic assessment of retinal disease images called a multi-view deep-broad learning network (MDBL-Net), which absorbs the advantages of deep learning and broad learning. MDBL-Net comprises a Multi-view and Multi-scale Feature Extraction (MMFE) module and a Multi-scale Aggregation (MA) block. The MMFE module extracts features of different scales by learning feature representations from multiple views, while the MA block fully aggregates multi-scale deep-broad features from low-level to high-level representations. Experiments were conducted on two public datasets, UCSD and OCT2017. Experiments were conducted on two public datasets, UCSD and OCT2017, and results demonstrate that MDBL-Net achieves high accuracy even with limited training data (only 1%) and significantly reduces training time compared to traditional deep learning models. Specifically, MDBL-Net achieved an accuracy of 96.93% on the UCSD dataset and 99.90% on the OCT2017 dataset, outperforming state-of-the-art models in both cases. These findings suggest that the proposed MDBL-Net approach holds great promise for the task of retinal disease screening and recognition.
Keywords
