Identifiable causal representation learning seeks to uncover the true causal relationships underlying a data generation process. In medical imaging, this presents opportunities to improve the generalisation and robustness of task-specific latent features. This work introduces the concept of grouping observations to learn identifiable representations for disease classification in chest X-rays via an end-to-end framework. Our experiments demonstrate that these causal representations improve performance across multiple classification tasks when grouping is used to enforce invariance with respect to race, sex, and imaging views.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Causal Representation Learning with Observational Grouping for CXR Classification

  • Rajat Rasal,
  • Avinash Kori,
  • Ben Glocker

摘要

Identifiable causal representation learning seeks to uncover the true causal relationships underlying a data generation process. In medical imaging, this presents opportunities to improve the generalisation and robustness of task-specific latent features. This work introduces the concept of grouping observations to learn identifiable representations for disease classification in chest X-rays via an end-to-end framework. Our experiments demonstrate that these causal representations improve performance across multiple classification tasks when grouping is used to enforce invariance with respect to race, sex, and imaging views.