mlsanityData to 7 Oct 2026

Not All NVFP4 QAT Recipes Are Equal: How Architecture and Scale Shape Model Quality for Anomaly Segmentation

arXiv 2605.27616 ↗First posted 26 May 2026cs.CV cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Real-time anomaly segmentation demands both high recall and efficient low-precision inference. We study the three-way interaction of model architecture, model scale, and FP4 quantization-aware training (QAT) recipe on a recall-critical brain tumor segmentation task, evaluating multiple architectures, scales, and QAT recipes under a unified protocol. We find that architecture choice has the largest impact on quantization robustness, with attention-based architectures showing remarkable resilience to recipe choice while CNN degrades under gradient-quantizing recipes at larger scales. At low capacity, FP4 can discretize softmax attention, but advanced QAT recipes prevent this collapse. At larger scales, advanced recipes mitigate gradient quantization noise that degrades CNN quality. Five-fold patient-level cross-validation confirms these findings are robust to data partition. Our results show that the Swin Transformer is robust to QAT recipe choice across all scales, making it the recommended architecture for FP4-quantized anomaly segmentation.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 21 of its 28 references that are in the map. First posted on arXiv 26 May 2026.

Builds on

21 of 28 references.

References · 21

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  2. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  3. 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
  4. 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
  5. 2021Vision Transformers for Dense Predictioncited by 1,789
  6. 2022How Do Vision Transformers Work?cited by 305
  7. 2022Transformers in Medical Imaging: A Surveycited by 245
  8. 2017Tversky loss function for image segmentation using 3D fully convolutional deep networkscited by 228
  9. 2018A Novel Focal Tversky loss function with improved Attention U-Net for lesion segmentationcited by 171
  10. 2021Unified Focal loss: Generalising Dice and cross entropy-based losses to handle class imbalanced medical image segmentationcited by 88
  11. 2019Association of genomic subtypes of lower-grade gliomas with shape features automatically extracted by a deep learning algorithmcited by 76
  12. 2024Why are Sensitive Functions Hard for Transformers?cited by 74
9 more
  1. 2018Bayesian Uncertainty Estimation for Batch Normalized Deep Networkscited by 70
  2. 2025Pretraining Large Language Models with NVFP4cited by 67
  3. 2018Towards Understanding Regularization in Batch Normalizationcited by 56
  4. 2024A Closer Look at AUROC and AUPRC under Class Imbalancecited by 56
  5. 2025FP4 All the Way: Fully Quantized Training of LLMscited by 37
  6. 2025Training LLMs with MXFP4cited by 36
  7. 2025Bridging the Gap Between Promise and Performance for Microscaling FP4 Quantizationcited by 30
  8. 2024Transformers Learn Low Sensitivity Functions: Investigations and Implicationscited by 13
  9. 2025Stochastic Rounding for LLM Training: Theory and Practicecited by 10

Built on by

Not cited yet.