Skip to content

parastak/MiniAZ-R-Reinforcement_Learning_Mathematical_Reasoning_solver

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

🧠 miniAZ-R: A Tiny Reinforced Math Solver Inspired by AZR (2025)

Inspired by the AZ-R paper “Absolute Zero: Reinforced Self-Play Reasoning with Zero Data” (2025) by Andrew Zhao Yiran Wu, Tong Wu, Quentin Xu, and collaborators.

This repository is not affiliated with the original authors, but serves as a practical miniature replication and experimental foundation toward that vision.

Screenshot

🎯 Goal

Train an RL agent from scratch to deduce symbolic mathematical equations through trial and error, guided by curriculum learning and self-play without supervised labels or solvers. From initial failure and debugging to achieving mastery on a curriculum of increasingly difficult mathematical tasks.

🔍 Features Implemented

✅ Actor-Critic agent with entropy & KL regularization

✅ Reward shaping and log_std tracking for exploration

✅ Curriculum learning pipeline (Easy → Medium → Hard tasks)

✅ Progressive reward & error convergence

✅ Evaluation dashboard & symbolic tagging

✅ Logging for interpretability and training insights

📸 Screenshots

Here are some results which showing a proper improvement in the agent reducing loss and growth in the reward signals

Initial agent learning results

Screenshot2 Screenshot1

latest agent policy results

Screenshot2 Screenshot1 Screenshot3

🔭 Future Work (toward full AZR)

  • Add post-hoc error correction
  • Integrate code-based symbolic execution
  • Diversify tasks to include chain-of-thought
  • Add multi-agent curriculum evolution
  • Enable multi-step program synthesis reasoning

🤝 Contribution & Citation

This project is not affiliated with the original AZR authors.
Inspired by research, built independently as a learning replica.

Open to collaborations, extensions, and feedback. Star ⭐ and fork if useful!


About

An implementation of an Actor-Critic agent that learns to solve algebraic equations of increasing difficulty through a self-play and curriculum learning framework, inspired by the 'Absolute Zero' research paper.

Topics

Resources

Stars

Watchers

Forks

Releases

Packages

Contributors

Languages