Improve Robustness of Eye Disease Detection by including Learnable Probabilistic Discrete Latent Variables into Machine Learning Models

1 code implementation • 21 Jan 2024 • Anirudh Prabhakaran, YeKun Xiao, Ching-Yu Cheng, Dianbo Liu

This study introduces a novel application of GFlowOut, leveraging the probabilistic framework of Generative Flow Networks (GFlowNets) to learn the posterior distribution over dropout masks, for the classification and analysis of ocular diseases using eye fundus images.

Decision Making