New Optimizer PsiLogic Enhances Adam for Stable AI Model Training

Ali Sultonov· July 21, 2026 View original

Summary

PsiLogic, a new optimizer, improves upon Adam by dynamically adjusting damping during training based on gradient stability, providing an implicit warmup and achieving superior performance in NLP and computer vision tasks. It includes a fair benchmark protocol and open-source implementation for verification.

Standard adaptive optimizers like Adam and AdamW apply a consistent update rule throughout the entire training process, regardless of whether the model is in an unstable initial phase or nearing convergence. Researchers have introduced PsiLogic, an enhanced optimizer that augments Adam with a dynamic "Active Cancellation Term." This term is controlled by a dual exponential moving average of scale-normalized gradient norms, effectively acting as a "chaos detector." When gradient statistics are unstable, PsiLogic increases damping, and as training stabilizes, this damping gradually diminishes, providing an automatic warmup without requiring manual scheduling. PsiLogic was rigorously evaluated against Adam, AdamW, and Lion using "FairBench," a new reproducible benchmark protocol. This protocol ensures fair comparisons through per-optimizer learning-rate sweeps, identical initialization across seeds, and statistical Welch t-tests. On an NVIDIA H100, PsiLogic demonstrated superior validation metrics in three out of four tested domains: NLP perplexity, ViT top-1 accuracy, and ResNet top-1 accuracy, showing statistically significant improvements over existing optimizers. While diffusion model performance was comparable, and some ResNet comparisons were numerically tied, the overall results highlight PsiLogic's effectiveness. The authors have released an open-source PyTorch implementation, the FairBench harness, and all raw data for independent verification.

Why it matters

AI engineers and researchers can leverage PsiLogic to achieve faster, more stable, and potentially more accurate training for deep learning models, reducing the need for manual hyperparameter tuning and improving model generalization.

How to implement this in your domain

  1. 1Review the open-source PyTorch implementation of PsiLogic and FairBench.
  2. 2Integrate PsiLogic into existing deep learning training pipelines, replacing Adam or AdamW.
  3. 3Conduct comparative experiments on current models to assess performance gains in terms of convergence speed and final accuracy.
  4. 4Utilize the FairBench protocol to ensure rigorous and reproducible evaluation of new optimizer implementations.
  5. 5Share findings within the engineering team to foster adoption and best practices for optimizer selection.

Who benefits

AI/ML DevelopmentSoftware DevelopmentResearch & AcademiaCloud Computing

Key takeaways

  • PsiLogic is a new optimizer that dynamically adjusts damping based on gradient stability.
  • It provides an implicit warmup, improving training stability and performance.
  • PsiLogic outperforms Adam and AdamW in several NLP and computer vision tasks.
  • An open-source implementation and fair benchmark protocol are available for adoption.

Original post by Ali Sultonov

"arXiv:2607.16268v1 Announce Type: new Abstract: Adaptive optimizers such as Adam and AdamW apply the same update rule regardless of whether training is in a chaotic early phase or near convergence. We introduce PsiLogic, an optimizer that augments Adam with a dynamic Active Cance…"

View on X

Originally posted by Ali Sultonov on X · view source

Want to go deeper?

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

Explore courses