• Home
  • Search
  • Stochastic Differential Equation Approach as Uncertainty‐Aware Feature Recalibration Module in Image Classification
  • https://doi.org/10.1002/ima.70131Copy DOI Icon

Stochastic Differential Equation Approach as Uncertainty‐Aware Feature Recalibration Module in Image Classification

Show More
  • Abstract
  • Literature Map
  • References
  • Similar Papers
Abstract

ABSTRACT Despite significant advancements in image classification, deep learning models struggle to accurately discern fine details in images, producing overly confident and imbalanced predictions for certain classes. These models typically employ feature recalibration techniques but do not account for the underlying uncertainty in predictions—particularly in complex sequential tasks like image classification. These uncertainties can significantly impact the reliability of subsequent analyses, potentially compromising accuracy across various applications. To address these limitations, we introduce the Stochastic Differential Equation Recalibration Module (SDERM), a novel approach designed to dynamically adjust the channel‐wise feature responses in convolutional neural networks. It integrates a stochastic differential equation (SDE) framework into a feature recalibration module to capture the inherent uncertainties in the data and its model predictions. To the best of our knowledge, our study is the first to explore the integration of SDE‐based feature recalibration modules in image classification. We build SDERM based on two interconnected networks—drift and diffusion network. The drift network serves as a deterministic component that approximates the predictive function of the model that systematically influences recalibrations of the predictions without considering the randomness. Concurrently, the diffusion network uses the Wiener process that captures the inherent uncertainties within the data and the network's predictions. We tested the classification accuracy of SDERM in ResNet50, ResNet101, and ResNet152 against other recalibration modules, including Squeeze‐Excitation (SE), Convolutional Block Attention Module (CBAM), Gather and Excite (GE), and Position‐Aware Recalibration Module (PARM), as well as the original Bottleneck architecture. Public image classification datasets were used, including CIFAR‐10, SVHN, FashionMNIST, and HAM10000, and their classification accuracies were evaluated using the F1 score. The proposed ResNetSDE architecture achieved state‐of‐the‐art F1 scores across four of five benchmark datasets. On Fashion‐MNIST, ResNetSDE attained an F1 score of 0.937 (CI: 0.932–0.941), outperforming all baseline recalibration methods by margins of 0.9%–1.3%. For CIFAR‐10 and CIFAR‐100, ResNetSDE achieved 0.886 (CI: 0.879–0.892) and 0.962 (CI: 0.958–0.965), respectively, surpassing ResNet‐GE and ResNet‐CBAM by 3.5% and 1.3%, respectively. ResNetSDE dominated SVHN with an F1 of 0.956 (CI: 0.953–0.958), a significant improvement over ResNet‐CBAM's 0.948 (CI: 0.945–0.951). While ResNet‐CBAM led on the class‐imbalanced HAM10000 (0.770, CI: 0.758–0.782), ResNetSDE remained competitive (0.768, CI: 0.749–0.786) since its consistent superiority—evidenced by narrow confidence intervals—validates its efficacy as a feature recalibration framework. Our experiments demonstrate that SDERM can outperform existing feature recalibration modules in image classification. The integration of SDERM to ResNet enables leveraging adaptability toward the stochasticity of each dataset at various depths of the architecture in image classification where uncertainty plays a fundamental role.

Similar Papers
  • Research Article

Brain-tumor Classification Using ResNet50 Enhanced with SE and CBAM Attention Mechanisms

  • Jan 01, 2026
  • Jordanian Journal of Computers and Information Technology
  • Nadia Shamsulddin Abdulsattar +1
  • Research Article

Optimization and Practice of Convolutional Neural Networks in Image Classification Tasks

  • Dec 21, 2024
  • International Journal of Computer Science and Information Technology
  • Lubin Liu
  • Research Article

Research on multi-scale convolutional neural network hand muscle strength prediction model improved based on convolutional attention module

  • Feb 25, 2025
  • Sheng wu yi xue gong cheng xue za zhi = Journal of biomedical engineering = Shengwu yixue gongchengxue zazhi
  • Yihao Du +4
  • Research Article

An OCT retinal image classification model based on improved ResNet-34 network

  • Nov 11, 2025
  • Biomedical Physics & Engineering Express
  • Zhenwei Li +5
  • Research Article
  • Citations11

An Efficient and Low-Complexity Transformer-Based Deep Learning Framework for High-Dynamic-Range Image Reconstruction.

  • Feb 28, 2025
  • Sensors (Basel, Switzerland)
  • Josue Lopez-Cabrejos +3
  • Research Article
  • Citations2

Glacial cirque identification based on Convolutional Neural Networks

  • Oct 24, 2024
  • Geomorphology
  • Dongxue Mao +4
  • Research Article
  • Citations2

A memory model for image recognition and classification based on convolutional neural network and Bayesian decision

  • Sep 01, 2017
  • SCIENTIA SINICA Technologica
  • Ying Jiang +3
  • PDF
  • Research Article
  • Citations30

Hybrid Convolutional Network Combining Multiscale 3D Depthwise Separable Convolution and CBAM Residual Dilated Convolution for Hyperspectral Image Classification

  • Oct 01, 2023
  • Remote Sensing
  • Yicheng Hu +2
  • Research Article
  • Citations12

Research on modulation recognition method of multi‐component radar signals based on deep convolution neural network

  • Jun 03, 2023
  • IET Radar, Sonar & Navigation
  • Chenxia Wan +2
  • Conference Article

Research on Automatic Extraction of Tobacco Plants Based on U2-Net of Convolutional Block Attention Module and UAV Visible Images

  • Aug 19, 2022
  • Chao Feng +5
  • Research Article
  • Citations6

Modeling Reactor Noise due to Rod and Thermal Vibrations with Thermal Feedback Using Stochastic Differential Equations

  • Oct 28, 2020
  • Nuclear Science and Engineering
  • Chen Dubi +1
  • Research Article
  • Citations1

A Novel Hybrid Approach: Capsule Network Enhanced with Residual Block and CBAM Module for Medical Image Classification

  • Apr 28, 2024
  • Proceedings of International Conference on Intelligent Systems and New Applications
  • Sumeyra Busra Sengul +1
  • Research Article
  • Citations10

Wind Turbine Bearing Failure Diagnosis Using Multi-Scale Feature Extraction and Residual Neural Networks with Block Attention

  • Oct 05, 2024
  • Actuators
  • Yuanqing Luo +5
  • Research Article
  • Citations42

A novel group squeeze excitation sparsely connected convolutional networks for SAR target classification

  • Jan 16, 2019
  • International Journal of Remote Sensing
  • Guoquan Huang +4
  • Research Article

Enhanced Breast Cancer Detection via Optimized Convolutional Block Attention Module (CBAM)

  • Jul 01, 2025
  • International Journal of Research Publication and Reviews
  • Mustapha Ismail +3
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.