Can we trace an AI model's answers back to its training data?
Surprisingly, even when we know exactly which documents taught a model a fact, attribution methods often fail to find them.
Our CATT benchmark investigates why and uncovers two simple fixes
catt.baulab.info🧵
Does this generalize? On FTRACE-TREX, we remove each method's top-ranked training documents and retrain. If those documents really taught the facts, accuracy on those facts should drop.
With our fixes, the accuracy drop is substantially larger for every method we test.
Try the demo to see how each method ranks training documents!
Paper: arxiv.org/abs/2610.40655
Demo: huggingface.co/spaces/catt-b…
Code: github.com/Lamsheeper/catt
Models & data: huggingface.co/catt-bench
Joint work with @ShengYuWang6, @kimiahmdh, Antonio Torralba and @davidbau!

