Beyond Softmax: Dual-Branch Sigmoid Architecture for Accurate Class Activation Maps
Not on the map. Why?
Abstract (arXiv)
Class Activation Mapping (CAM) and its extensions have become indispensable tools for visualizing the evidence behind deep network predictions. However, by relying on a final softmax classifier, these methods suffer from two fundamental distortions: additive logit shifts that arbitrarily bias importance scores, and sign collapse that conflates excitatory and inhibitory features. We propose a simple, architecture-agnostic dual-branch sigmoid head that decouples localization from classification. Given any pretrained model, we clone its classification head into a parallel branch ending in per-class sigmoid outputs, freeze the original softmax head, and fine-tune only the sigmoid branch with class-balanced binary supervision. At inference, softmax retains recognition accuracy, while class evidence maps are generated from the sigmoid branch -- preserving both magnitude and sign of feature contributions. Our method integrates seamlessly with most CAM variants and incurs negligible overhead. Extensive evaluations on fine-grained tasks (CUB-200-2011, Stanford Cars) and WSOL benchmarks (ImageNet-1K, OpenImages30K) show improved explanation fidelity and consistent Top-1 Localization gains -- without any drop in classification accuracy. Code is available at https://github.com/finallyupper/beyond-softmax.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 26 of its 31 references that are in the map. First posted on arXiv 5 Nov 2025.
Builds on
26 of 31 references.
References · 26
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,498
- 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,733
- 2017Squeeze-and-Excitation Networkscited by 4,789
- 2014Going Deeper with Convolutionscited by 4,241
- 2017Axiomatic Attribution for Deep Networkscited by 3,384
- 2015Learning Deep Features for Discriminative Localizationcited by 2,436
- 2013Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Mapscited by 2,371
- 2017Learning Important Features Through Propagating Activation Differencescited by 1,458
- 2017SmoothGrad: removing noise by adding noisecited by 1,109
- 2017Grad-CAM++: Improved Visual Explanations for Deep Convolutional Networkscited by 939
14 more
- 2018RISE: Randomized Input Sampling for Explanation of Black-box Modelscited by 838
- 2020Quantifying Attention Flow in Transformerscited by 775
- 2019Score-CAM: Score-Weighted Visual Explanations for Convolutional Neural Networkscited by 582
- 2019Understanding Deep Networks via Extremal Perturbations and Smooth Maskscited by 259
- 2017Hide-and-Seek: Forcing a Network to be Meticulous for Weakly-supervised Object and Action Localizationcited by 207
- 2017Object Region Mining with Adversarial Erasing: A Simple Classification to Semantic Segmentation Approachcited by 201
- 2020Eigen-CAM: Class Activation Map using Principal Componentscited by 189
- 2018Adversarial Complementary Learning for Weakly Supervised Object Localizationcited by 166
- 2020Axiom-based Grad-CAM: Towards Accurate Visualization and Explanation of CNNscited by 151
- 2019Attention-based Dropout Layer for Weakly Supervised Object Localizationcited by 125
- 2021TS-CAM: Token Semantic Coupled Attention Map for Weakly Supervised Object Localizationcited by 123
- 2019Large-scale interactive object segmentation with human annotatorscited by 110
- 2020Evaluating Weakly Supervised Object Localization Methods Rightcited by 105
- 2025Finer-CAM: Spotting the Difference Reveals Finer Details for Visual Explanationcited by 15
Built on by
Not cited yet.