Identifying Subgroups in Biomedical Datasets using Data Attribution @MITCBMM
Identifying Subgroups in Biomedical Datasets using Data Attribution  @MITCBMM
Uploaded October 2024 | Updated September 2026, 1 week ago
Djuna von Maydell, MIT

Understanding how training data influences model predictions ("data attribution") is an active area of machine learning research. In this tutorial, we will introduce a data attribution method (datamodels: gradientscience.org/datamodels-1/) and explore how it can be applied in the life sciences to identify meaningful subgroups in biomedical datasets, such as disease subtypes. We will begin with a simple example from image classification (CIFAR10), offering a step-by-step guide to demonstrate how the data attribution method works in practice. Since the approach involves training thousands of lightweight classifiers, we will focus on strategies for fast and efficient model training. Next, we will explore its applications in biomedical science, with a focus on single-cell and genetic datasets, highlighting the biological insights gained from applying this computational approach. The tutorial will conclude with an interactive, hands-on session using Google Colab, where participants can apply the techniques themselves and explore the approach further. This session is designed to be accessible to participants of all coding and machine learning experience levels—whether you're new to machine learning or curious about its intersection with biomedical applications.

Slides: drive.google.com/file/d/1qGahNYBUnThba07D2D9gZTviiU_kOedF/view?usp=sharing

Github repository of tutorial code: github.com/djunamay/datamodels_tutorial

Code with outputs: colab.research.google.com/drive/1u2jZzWs7SVT6kj-O8rMsUphHfvyeqnHh?usp=sharing

Code no outputs:
colab.research.google.com/drive/1lwl7-Xsc7lg9bTg97hEEqPt54x-J1qeU?usp=sharing
Identifying Subgroups in Biomedical Datasets using Data AttributionHow do humans sense wetnessConveying Tasks to Computers: How Machine Learning Can HelpStatistical learning in human sensorimotor controlPhysical and Social Human-Robot Interaction with the iCub HumanoidTowards complex language in partially observed environmentsEfficient representation, learning, and planning through abstraction: clustering cognitive spaces...InvestigacionWobbling, drooping, bouncing - Visual perception of materials and their propertiesMicroscopia de Flourescencia: El microscopio como una herramienta de trabajoTutorial: Recurrent neural networks for cognitive neuroscienceDo rats dream?
MITCBMM |

Identifying Subgroups in Biomedical Datasets using Data Attribution

SHARE TO X SHARE TO REDDIT SHARE TO FACEBOOK WALLPAPER