Endre søk
RefereraExporteraLink to record
Permanent link

Direct link
Referera
Referensformat
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Annet format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annet 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. KTH Royal Institute of Technology, Sweden.
KTH, Centra, Science for Life Laboratory, SciLifeLab. KTH, Skolan för elektroteknik och datavetenskap (EECS), Intelligenta system, Robotik, perception och lärande, RPL. KTH, Centra, SeRC - Swedish e-Science Research Centre. KTH Royal Institute of Technology, Sweden.ORCID-id: 0000-0001-5211-6388
2024 (engelsk)Inngår i: Proceedings of the 5th Conference on Health, Inference, and Learning, CHIL 2024, ML Research Press , 2024, s. 155-168Konferansepaper, Publicerat paper (Fagfellevurdert)
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 ex-plainability 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.

sted, utgiver, år, opplag, sider
ML Research Press , 2024. s. 155-168
HSV kategori
Identifikatorer
URN: urn:nbn:se:kth:diva-353944Scopus ID: 2-s2.0-85203788338OAI: oai:DiVA.org:kth-353944DiVA, id: diva2:1901020
Konferanse
5th Annual Conference on Health, Inference, and Learning, CHIL 2024, New York, United States of America, Jun 27 2024 - Jun 28 2024
Merknad

QC 20240926

Tilgjengelig fra: 2024-09-25 Laget: 2024-09-25 Sist oppdatert: 2025-02-01bibliografisk kontrollert

Open Access i DiVA

Fulltekst mangler i DiVA

Scopus

Person

Nilsson, AlfredAzizpour, Hossein

Søk i DiVA

Av forfatter/redaktør
Nilsson, AlfredAzizpour, Hossein
Av organisasjonen

Søk utenfor DiVA

GoogleGoogle Scholar

urn-nbn

Altmetric

urn-nbn
Totalt: 193 treff
RefereraExporteraLink to record
Permanent link

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