OriginBlame Tracks Data Provenance for AI Unlearning.

Haolin Xue· July 16, 2026 View original

Summary

OriginBlame is a novel system providing record- and token-level data provenance for AI training datasets, enabling precise identification of data for unlearning requests. It significantly reduces over-deletion compared to file-level systems and integrates with minimal throughput overhead in data processing pipelines.

When data contributors request the removal of their information, AI model trainers face a significant challenge: existing provenance systems typically operate at a coarse file or dataset level, leading to excessive data deletion during unlearning. To address this, a new system called OriginBlame (ob) has been developed. OriginBlame offers granular data provenance at both the record and token levels. It propagates author identity through data processing pipelines, allowing for deterministic queries that precisely identify which training records belong to a specific author. This capability enables the creation of accurate "forget sets" for unlearning algorithms. Evaluations show that OriginBlame drastically reduces over-deletion (from 101x to 1.3x) compared to dataset-level methods, with only a modest throughput overhead (1.3-4.0% for HuggingFace, 2.1-19.0% for Datatrove) during integration. For a 1.7B model, provenance-based forget sets improved unlearning effectiveness by 42% over random baselines.

Why it matters

This tool is crucial for AI developers and data privacy officers, enabling compliance with data removal requests (e.g., GDPR "right to be forgotten") by providing precise data provenance for effective model unlearning.

How to implement this in your domain

  1. 1Assess current data provenance capabilities within your AI training pipelines.
  2. 2Investigate integrating OriginBlame or similar record-level provenance systems for compliance and data management.
  3. 3Develop protocols for handling data removal requests using precise forget sets generated by provenance tools.
  4. 4Benchmark the overhead of provenance tracking on your specific data processing workflows.
  5. 5Educate data governance and legal teams on the capabilities of granular data provenance for AI unlearning.

Who benefits

AI/ML DevelopmentData GovernanceLegal/ComplianceCloud ServicesSocial Media

Key takeaways

  • OriginBlame provides record- and token-level data provenance for AI datasets.
  • It enables precise identification of data for unlearning requests.
  • The system significantly reduces over-deletion compared to file-level methods.
  • Integration adds minimal throughput overhead to data processing.

Original post by Haolin Xue

"arXiv:2607.13037v1 Announce Type: new Abstract: When a data contributor requests removal, model trainers face a practical gap: unlearning algorithms require a forget set, yet no tool can locate which training records belong to a given author. Existing provenance systems operate a…"

View on X

Originally posted by Haolin Xue on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses