Skip to content

Bergson: An Open Source Library for Data Attribution

Jun 2026 · arXiv.org · Vol abs/2606.11660 · 0 citations · 59 references
Computer Science

TL;DR

Bergson is an open source library that aims to enable faster progress in the field by providing a host of techniques that scale to very large language models and pre-training datasets, and provides quality of life tools for researchers.

Abstract

Data attribution is a promising field in interpretability that aims to explain model behavior through the influence of its training data, with applications including debugging undesirable model behavior and training dataset curation. However, significant engineering effort is required to perform it at scale, and many cutting edge techniques lack open-source tooling and support. Bergson is an open source library that aims to enable faster progress in the field by providing a host of techniques that scale to very large language models and pre-training datasets. The library natively supports on-disk gradient stores and multi-node distributed training, and provides quality of life tools for researchers. Finally, we introduce the first open-source implementations of three leading data attribution methods: MAGIC, SOURCE, and TrackStar. The library is available at https://github.com/EleutherAI/bergson .

View source

Similar papers

#artificial intelligence Preprint Sep 2026

dattri-LLM: A Unified and Efficient Library for Training Data Attribution at LLM Scale

Training data attribution (TDA) estimates the contribution of individual training examples to model outputs. Most scalable TDA methods rely on per-example gradients, whose computation and use at LLM scale pose challenges in efficiency, compatibility, and extensibility. We introduce dattri-LLM, a TDA library that makes...

Shi-Xuan Liu, Tong-Li Zhou, Jun-Wei Deng et al. · 0 citations
#machine learning Preprint Sep 2026

WOMBAT: Whitebox Oracle for Molecular Benchmarking and Attribution Testing

When a graph neural network (GNN) explainer produces an unexpected attribution on a molecule, the attribution alone cannot reveal whether the explainer has failed or the model has learned a shortcut. We introduce WOMBAT, a benchmark of 14 whitebox GNNs, each with message-passing weights set by hand to detect a specific...

Dominik Matuszek, Bartosz Zielinski, T. Danel et al. · 0 citations
#machine learning Preprint Sep 2026

Traceprop: Training Data Attribution in a Single Pass

The contribution here is systems, not a new estimator: LogIX's attribution quality, in one pass instead of two, is systems, not a new estimator: LogIX's attribution quality, in one pass instead of two.

Amit Nautiyal · 0 citations
#artificial intelligence Preprint Sep 2026

Form Over Content In Gradient-Based Data Attribution Methods

Data attribution methods using gradient similarity are widely used to analyze and select training data for large language models, but what gradient similarity actually measures is debated. Some interpret it as identifying task-relevant skills, while other work reports that surface form is the main factor. We resolve th...

Sunwoo Kim, Seokwon Jung, Sohyung Kim et al. · 1 citation

Can LLMs Serve as a Data Error Detection Engine? Trade-offs in Accuracy, Cost, and Hallucination Across Datasets

These findings endorse the viability of LLMs as a feasible alternative for data error detection; nonetheless, challenges such as hallucination and cost aspects remain, especially regarding the resources necessary for the effective implementation of these models in practical applications.

Maximilian Plazotta, Meike Klettke · 0 citations

Related blog posts

Microsoft Research Blog Oct 12, 2026

Simulating student learning for better AI tutoring

StudentSim is a research system for evaluating and improving AI tutoring tools before they are tested with learners. It models how individuals may engage with tutors, giving feedback to improve tutor performance across subjects like chess, writing, and mathematics. The post Simulating student learning for better AI tutoring appeared first on Microsoft Research.

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.