[P] MCGrad: fix calibration of your ML model in subgroups
We’re open-sourcing MCGrad, a Python package for multicalibration–developed and deployed in production at Meta. This work will also be presented at KDD 2026.
The Problem: A model can be globally calibrated yet significantly miscalibrated within identifiable subgroups or feature intersections (e.g., "users in region X on mobile devices"). Multicalibration aims to ensure reliability across such subpopulations.
The Solution: MCGrad reformulates multicalibration using gradient boosted decision trees. At each step, a lightweight booster learns to predict residual miscalibration of the base model given the features, automatically identifying and correcting miscalibrated regions. The method scales to large datasets, and uses early stopping to preserve predictive performance. See our tutorial for a live demo.
Key Results: Across 100+ production models at meta, MCGrad improved log loss and PRAUC on 88% of them while substantially reducing subgroup calibration error.
Links:
- Repo: https://github.com/facebookincubator/MCGrad/
- Docs: https://mcgrad.dev/
- Paper: https://arxiv.org/abs/2509.19884
Install via pip install mcgrad or via conda. Happy to answer questions or discuss details.
[link] [comments]
Want to read more?
Check out the full article on the original site