Learning to Stop: Reinforcement Learning for Efficient Patient-Level Echocardiographic Classification
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866911173132156928 |
|---|---|
| author | Kim, Woo-Jin Cho Oliveira, Jorge Beqiri, Arian Thorley, Alex Strom, Jordan O'Driscoll, Jamie Sharma, Rajan Slivnick, Jeremy Lang, Roberto Gomez, Alberto Chartsias, Agisilaos |
| author_facet | Kim, Woo-Jin Cho Oliveira, Jorge Beqiri, Arian Thorley, Alex Strom, Jordan O'Driscoll, Jamie Sharma, Rajan Slivnick, Jeremy Lang, Roberto Gomez, Alberto Chartsias, Agisilaos |
| contents | Guidelines for transthoracic echocardiographic examination recommend the acquisition of multiple video clips from different views of the heart, resulting in a large number of clips. Typically, automated methods, for instance disease classifiers, either use one clip or average predictions from all clips. Relying on one clip ignores complementary information available from other clips, while using all clips is computationally expensive and may be prohibitive for clinical adoption.
To select the optimal subset of clips that maximize performance for a specific task (image-based disease classification), we propose a method optimized through reinforcement learning. In our method, an agent learns to either keep processing view-specific clips to reduce the disease classification uncertainty, or stop processing if the achieved classification confidence is sufficient. Furthermore, we propose a learnable attention-based aggregation method as a flexible way of fusing information from multiple clips. The proposed method obtains an AUC of 0.91 on the task of detecting cardiac amyloidosis using only 30% of all clips, exceeding the performance achieved from using all clips and from other benchmarks. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2509_19694 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Learning to Stop: Reinforcement Learning for Efficient Patient-Level Echocardiographic Classification Kim, Woo-Jin Cho Oliveira, Jorge Beqiri, Arian Thorley, Alex Strom, Jordan O'Driscoll, Jamie Sharma, Rajan Slivnick, Jeremy Lang, Roberto Gomez, Alberto Chartsias, Agisilaos Computer Vision and Pattern Recognition Guidelines for transthoracic echocardiographic examination recommend the acquisition of multiple video clips from different views of the heart, resulting in a large number of clips. Typically, automated methods, for instance disease classifiers, either use one clip or average predictions from all clips. Relying on one clip ignores complementary information available from other clips, while using all clips is computationally expensive and may be prohibitive for clinical adoption. To select the optimal subset of clips that maximize performance for a specific task (image-based disease classification), we propose a method optimized through reinforcement learning. In our method, an agent learns to either keep processing view-specific clips to reduce the disease classification uncertainty, or stop processing if the achieved classification confidence is sufficient. Furthermore, we propose a learnable attention-based aggregation method as a flexible way of fusing information from multiple clips. The proposed method obtains an AUC of 0.91 on the task of detecting cardiac amyloidosis using only 30% of all clips, exceeding the performance achieved from using all clips and from other benchmarks. |
| title | Learning to Stop: Reinforcement Learning for Efficient Patient-Level Echocardiographic Classification |
| topic | Computer Vision and Pattern Recognition |
| url | https://arxiv.org/abs/2509.19694 |