qurAI
- Location
- Netherlands
- Website
- https://qurai.amsterdam/
- Editors
Activity Overview
Age-related macular degeneration (AMD) Staging in Optical Coherence Tomography (OCT) with UBIX for Increased Reliability
AlgorithmDeep learning models for optical coherence tomography (OCT) classification often perform well on data from scanners that were also used during training. However, when these models are applied to data from different vendors, their performance tends to drop substantially. Artifacts that only occur within scans from specific scanners are major causes of this poor generalizability. We aimed to improve this generalizability of deep learning models for OCT classifi- cation. To reduce the effect of vendor-specific artifacts, we propose Uncertainty-Based Instance eXclusion (UBIX), of which we define a hard and a soft variant. UBIX aims to suppress the contributions of B-scans with unseen artifacts to the final OCT-level outputs. Suppression is based on out-of-distribution detection of B-scans, which are instances in our multiple instance learning approach.