1 min readfrom Machine Learning

Optimizing Transformer model size & inference beyond FP16 + ONNX (pruning/graph opt didn’t help much) [P]

Our take

In the quest to optimize transformer-based neural networks for model size and inference speed, I’ve encountered a plateau after implementing several strategies. I've successfully converted weights to FP16, achieving a 2× size reduction, and utilized ONNX Runtime for speed enhancements. Despite attempts at unstructured and structured pruning, along with ONNX graph optimizations, I’m still at approximately 162 MB per model. I’m now considering options like low-rank factorization, aggressive quantization, and knowledge distillation.

Hi everyone, I’ve been working on optimizing a transformer-based neural network for both inference speed and model size, but I feel like I’ve hit a plateau and would appreciate some guidance. So far I’ve converted weights to FP16 (about 2× size reduction), exported and optimized with ONNX Runtime for inference speed, and tried both unstructured and structured pruning as well as ONNX graph optimizations, but none of these gave significant additional gains, and I’m still around ~162 MB per model. At this point I’m considering next steps like low-rank factorization (SVD/LoRA-style compression), more aggressive quantization (INT8/INT4 like GPTQ, AWQ, or SmoothQuant), knowledge distillation into a smaller student model, or more hardware/runtime-specific optimizations like TensorRT or FlashAttention, but I’m not sure which of these actually gives meaningful real-world improvements after FP16 + pruning. I’d really appreciate advice on what approaches tend to work best in practice for transformer compression beyond what I’ve already tried, and whether low-rank methods are actually effective post-training or if distillation/quantization is usually the only real win at this stage.

submitted by /u/Fragrant_Rate_2583
[link] [comments]

Read on the original site

Open the publisher's page for the full experience

View original article

Related Articles