ATrain
ATrain · Team Ajsel Budlla
Submitted to The Technical AI Governance Challenge. Sprint projects are early-stage work by participants, not Apart Research publications.
ATrain is a system that lets us prove, in a secure and transparent way, how much compute a model actually used during training. It logs training metrics, estimates computational usage, and cryptographically signs the data so anyone can verify it hasn’t been tampered with. We built a simple dashboard in Colab where you can start a training run and generate graphs showing whether the model stayed within regulatory thresholds. Essentially, it turns AI training from a trust-based process into one that can be independently verified without exposing your model weights or sensitive data. This could help labs and regulators ensure safe scaling of AI systems.

Reviews
The paper correctly identifies that current approaches in compute governance rely on voluntary disclosure rather than verifiable evidence, and helps solve one part of the problem of verification. The emphasis on an accessible Colab-based dashboard is a good design choice for a prototype, lowering the barrier for non-technical stakeholders to interact with attestation concepts, which has value for building intuition about what compute governance tooling could look like. However, the system is at present self-attesting: the training process reports its own compute, then cryptographically signs its own report. The signature proves logs weren't altered after signing, but says nothing about whether they were accurate in the first place, with a dishonest actor able to produce false logs and sign those. The paper claims to transform training "from trust-based to proof-based," but it relocates the trust assumption up the chain rather than eliminating it. This is a useful first layer in what would need to be a multi-layered verification stack, but other components, such as a TEE or hardware-level measurement providing an external root of trust, would still be necessary for cryptographic reporting of compute usage to be reliable.
Read full reviewShow less
This is well built and easy to understand. The dashboard + workflow is clean.
But the core idea (hash + sign logs to prove integrity) is a pretty standard pattern. So the novelty is limited. Also: it mainly proves “the log wasn’t changed,” not “the training run was honest.” If the environment lies, a signed log can still be fake.
To make this more impactful, I want a clear threat model and a real path to hardware-backed / remote attestation. Right now it’s good compliance tooling, but not a new safety mechanism.
Cite this project
@misc{atrain2026atrain,
title = {{ATrain}},
author = {ATrain},
year = {2026},
month = feb,
note = {Submitted to The Technical AI Governance Challenge, an Apart Research Sprint},
howpublished = {\url{https://apartresearch.com/sprints/projects/atrain-mvbo}},
url = {https://apartresearch.com/sprints/projects/atrain-mvbo}
}More from The Technical AI Governance Challenge
- 1st placeView project: LidaSim: Testing AI Policies With Persona-Based Simulations
LidaSim: Testing AI Policies With Persona-Based Simulations
Lida Safety
We simulate well-known figures in AI and politics with agents, scraping large amounts of data to get realistic simulations. Then, we test questions and proposed policies against these public figures, to see which …
- 2nd placeView project: Markov Chain Lock Watermarking: Provably Secure Authentication for LLM Outputs
Markov Chain Lock Watermarking: Provably Secure Authentication for LLM Outputs
MCL
We present Markov Chain Lock (MCL) watermarking, a cryptographically secure framework for authenticating LLM outputs. MCL constrains token generation to follow a secret Markov chain over SHA-256 vocabulary partitions. …
- 3rd placeView project: Political Intelligence for AI Safety: The AI Risk Attitudes Survey (AIRAS)
Political Intelligence for AI Safety: The AI Risk Attitudes Survey (AIRAS)
AIRAS
The AI safety and governance community is making progress on defining red lines around existential risk from advanced AI systems, and building verification infrastructure to support this objective. However, this is only …