Tri-Accel: Curvature-Aware Precision-Adaptive and Memory-Elastic Optimization for Efficient GPU Usage
Not on the map. Why?
Abstract (arXiv)
Deep neural networks are increasingly bottlenecked by the cost of optimization, both in terms of GPU memory and compute time. Existing acceleration techniques, such as mixed precision, second-order methods, and batch size scaling, are typically used in isolation. We present Tri-Accel, a unified optimization framework that co-adapts three acceleration strategies along with adaptive parameters during training: (1) Precision-Adaptive Updates that dynamically assign mixed-precision levels to layers based on curvature and gradient variance; (2) Sparse Second-Order Signals that exploit Hessian/Fisher sparsity patterns to guide precision and step size decisions; and (3) Memory-Elastic Batch Scaling that adjusts batch size in real time according to VRAM availability. On CIFAR-10 with ResNet-18 and EfficientNet-B0, Tri-Accel achieves up to 9.9% reduction in training time and 13.3% lower memory usage, while improving accuracy by +1.1 percentage points over FP32 baselines. Tested on CIFAR-10/100, our approach demonstrates adaptive learning behavior, with efficiency gradually improving over the course of training as the system learns to allocate resources more effectively. Compared to static mixed-precision training, Tri-Accel maintains 78.1% accuracy while reducing memory footprint from 0.35GB to 0.31GB on standard hardware. The framework is implemented with custom Triton kernels, whose hardware-aware adaptation enables automatic optimization without manual hyperparameter tuning, making it practical for deployment across diverse computational environments. This work demonstrates how algorithmic adaptivity and hardware awareness can be combined to improve scalability in resource-constrained settings, paving the way for more efficient neural network training on edge devices and cost-sensitive cloud deployments.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 36 of its 55 references that are in the map. First posted on arXiv 23 Aug 2025.
Builds on
36 of 55 references.
References · 36
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,868
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
- 2017Mixed Precision Trainingcited by 1,156
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 2019Large Batch Optimization for Deep Learning: Training BERT in 76 minutescited by 674
- 2017Large Batch Training of Convolutional Networkscited by 535
- 2021Sparsity in Deep Learning: Pruning and growth for efficient inference and training in neural networkscited by 527
- 2014New insights and perspectives on the natural gradient methodcited by 443
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
- 2015Deep Learning with Limited Numerical Precisioncited by 376
24 more
- 2017Don't Decay the Learning Rate, Increase the Batch Sizecited by 320
- 2018An Empirical Model of Large-Batch Trainingcited by 281
- 2021Accelerating Sparse Deep Neural Networkscited by 223
- 2018Training Deep Neural Networks with 8-bit Floating Point Numberscited by 164
- 2019Practical Deep Learning with Bayesian Principlescited by 158
- 2018Scalable Methods for 8-bit Training of Neural Networkscited by 142
- 2016A Kronecker-factored approximate Fisher matrix for convolution layerscited by 141
- 2020Memory-Efficient Pipeline-Parallel DNN Trainingcited by 134
- 2018PipeDream: Fast and Efficient Pipeline Parallel DNN Trainingcited by 117
- 2020HetPipe: Enabling Large DNN Training on (Whimpy) Heterogeneous GPU Clusters through Integration of Pipelined Model Parallelism and Data Parallelismcited by 46
- 2018Mixed Precision Training of Convolutional Neural Networks using Integer Operationscited by 45
- 2018A Progressive Batching L-BFGS Method for Machine Learningcited by 45
- 2024Can LLMs Improve Multimodal Fact-Checking by Asking Relevant Questions?cited by 17
- 2024Increasing Both Batch Size and Learning Rate Accelerates Stochastic Gradient Descentcited by 11
- 2024A Semi-supervised Fake News Detection using Sentiment Encoding and LSTM with Self-Attentioncited by 8
- 2020Weight Update Skipping: Reducing Training Time for Artificial Neural Networkscited by 6
- 2025A Multimodal Physics-Informed Neural Network Approach for Mean Radiant Temperature Modelingcited by 6
- 2022EasyScale: Accuracy-consistent Elastic Training for Deep Learningcited by 3
- 2025Sentiment and Social Signals in the Climate Crisis: A Survey on Analyzing Social Media Responses to Extreme Weather Eventscited by 3
- 2024CUDLE: Learning Under Label Scarcity to Detect Cannabis Use in Uncontrolled Environmentscited by 2
- 2025MID-L: Matrix-Interpolated Dropout Layer with Layer-wise Neuron Selectioncited by 2
- 2025MNIST-Gen: A Modular MNIST-Style Dataset Generation Using Hierarchical Semantics, Reinforcement Learning, and Category Theorycited by 2
- 2020SQWA: Stochastic Quantized Weight Averaging for Improving the Generalization Capability of Low-Precision Deep Neural Networkscited by 1
- 2024A Physics-Informed Machine Learning Approach for Solving Distributed Order Fractional Differential Equationscited by 1
Built on by
Not cited yet.