kth.sePublikationer KTH
Driftstörningar
Det förekommer återkommande driftstörningar på några av sök-portalerna på grund av hög belastning. Vi arbetar på att lösa problemet, ni kan tillfälligt mötas av ett felmeddelande.
Ändra sökning
RefereraExporteraLänk till posten
Permanent länk

Direktlänk
Referera
Referensformat
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Annat format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annat språk
Fler språk
Utmatningsformat
  • html
  • text
  • asciidoc
  • rtf
Regularizing and Interpreting Vision Transformers by Patch Selection on Echocardiography Data
KTH, Skolan för kemi, bioteknologi och hälsa (CBH), Genteknologi.ORCID-id: 0000-0002-3181-3800
KTH, Skolan för elektroteknik och datavetenskap (EECS), Intelligenta system, Robotik, perception och lärande, RPL.ORCID-id: 0000-0001-5211-6388
2024 (Engelska)Ingår i: CONFERENCE ON HEALTH, INFERENCE, AND LEARNING / [ed] Pollard, T Choi, E Singhal, P Hughes, M Sizikova, E Mortazavi, B Chen, I Wang, F Sarker, T McDermott, M Ghassemi, M, The Journal of Machine Learning Research (JMLR) , 2024, Vol. 248, s. 155-168Konferensbidrag, Publicerat paper (Refereegranskat)
Abstract [en]

This work introduces a novel approach to model regularization and explanation in Vision Transformers (ViTs), particularly beneficial for small-scale but high-dimensional data regimes, such as in healthcare. We introduce stochastic embedded feature selection in the context of echocardiography video analysis, specifically focusing on the EchoNet-Dynamic dataset for the prediction of Left Ventricular Ejection Fraction (LVEF). Our proposed method, termed Gumbel Video Vision-Transformers (G-ViTs), augments Video Vision-Transformers (V-ViTs), a performant transformer architecture for videos with Concrete Autoencoders (CAEs), a common dataset-level feature selection technique, to enhance V-ViT's generalization and interpretability. The key contribution lies in the incorporation of stochastic token selection individually for each video frame during training. Such token selection regularizes the training of V-ViT, improves its interpretability, and is achieved by differentiable sampling of categoricals using the Gumbel-Softmax distribution. Our experiments on EchoNet-Dynamic demonstrate a consistent and notable regularization effect. The G-ViT model outperforms both a random selection baseline and standard V-ViT. The G-ViT is also compared against recent works on EchoNet-Dynamic where it exhibits state-of-the-art performance among end-to-end learned methods. Finally, we explore model explainability by visualizing selected patches, providing insights into how the G-ViT utilizes regions known to be crucial for LVEF prediction for humans. This proposed approach, therefore, extends beyond regularization, offering enhanced interpretability for ViTs.

Ort, förlag, år, upplaga, sidor
The Journal of Machine Learning Research (JMLR) , 2024. Vol. 248, s. 155-168
Serie
Proceedings of Machine Learning Research, ISSN 2640-3498
Nationell ämneskategori
Datorgrafik och datorseende
Identifikatorer
URN: urn:nbn:se:kth:diva-357514ISI: 001347132700011OAI: oai:DiVA.org:kth-357514DiVA, id: diva2:1919587
Konferens
5th Annual Conference on Health, Inference, and Learning (CHIL), JUN 27-28, 2024, New York, NY
Anmärkning

QC 20241209

Tillgänglig från: 2024-12-09 Skapad: 2024-12-09 Senast uppdaterad: 2025-02-07Bibliografiskt granskad

Open Access i DiVA

Fulltext saknas i DiVA

Person

Nilsson, AlfredAzizpour, Hossein

Sök vidare i DiVA

Av författaren/redaktören
Nilsson, AlfredAzizpour, Hossein
Av organisationen
GenteknologiRobotik, perception och lärande, RPL
Datorgrafik och datorseende

Sök vidare utanför DiVA

GoogleGoogle Scholar

urn-nbn

Altmetricpoäng

urn-nbn
Totalt: 181 träffar
RefereraExporteraLänk till posten
Permanent länk

Direktlänk
Referera
Referensformat
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Annat format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annat språk
Fler språk
Utmatningsformat
  • html
  • text
  • asciidoc
  • rtf