Learning Disease State from Noisy Ordinal Disease Progression Labels

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Schmidt, Gustav, Heidrich, Holger, Berens, Philipp, Müller, Sarah
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915432976351232
author Schmidt, Gustav
Heidrich, Holger
Berens, Philipp
Müller, Sarah
author_facet Schmidt, Gustav
Heidrich, Holger
Berens, Philipp
Müller, Sarah
contents Learning from noisy ordinal labels is a key challenge in medical imaging. In this work, we ask whether ordinal disease progression labels (better, worse, or stable) can be used to learn a representation allowing to classify disease state. For neovascular age-related macular degeneration (nAMD), we cast the problem of modeling disease progression between medical visits as a classification task with ordinal ranks. To enhance generalization, we tailor our model to the problem setting by (1) independent image encoding, (2) antisymmetric logit space equivariance, and (3) ordinal scale awareness. In addition, we address label noise by learning an uncertainty estimate for loss re-weighting. Our approach learns an interpretable disease representation enabling strong few-shot performance for the related task of nAMD activity classification from single images, despite being trained only on image pairs with ordinal disease progression labels.
format Preprint
id arxiv_https___arxiv_org_abs_2503_10440
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning Disease State from Noisy Ordinal Disease Progression Labels
Schmidt, Gustav
Heidrich, Holger
Berens, Philipp
Müller, Sarah
Computer Vision and Pattern Recognition
Machine Learning
Learning from noisy ordinal labels is a key challenge in medical imaging. In this work, we ask whether ordinal disease progression labels (better, worse, or stable) can be used to learn a representation allowing to classify disease state. For neovascular age-related macular degeneration (nAMD), we cast the problem of modeling disease progression between medical visits as a classification task with ordinal ranks. To enhance generalization, we tailor our model to the problem setting by (1) independent image encoding, (2) antisymmetric logit space equivariance, and (3) ordinal scale awareness. In addition, we address label noise by learning an uncertainty estimate for loss re-weighting. Our approach learns an interpretable disease representation enabling strong few-shot performance for the related task of nAMD activity classification from single images, despite being trained only on image pairs with ordinal disease progression labels.
title Learning Disease State from Noisy Ordinal Disease Progression Labels
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2503.10440