1. Introduction
1.1. Motivation and Problem Statement
The optimization of deep neural networks under imperfect supervision conditions presents fundamental algorithmic challenges that extend beyond empirical performance metrics. When training data contains label noise—whether from annotation ambiguity, inter-rater disagreement, or inherent task uncertainty, standard gradient-based optimization can lead to memorization of incorrect labels, degraded generalization, and unstable convergence dynamics [
1,
2].
This challenge is particularly acute for lightweight neural network architectures, where reduced model capacity limits the network’s ability to simultaneously fit clean patterns while rejecting noisy supervision signals. The theoretical analysis by Bubeck and Sellke [
3] established a “universal law of robustness” demonstrating that achieving smooth interpolation requires approximately d times more parameters than mere interpolation, where d is the data dimension. For resource-constrained deployments requiring compact models, this creates a fundamental tension between parameter efficiency and optimization robustness. Crucially, this theoretical result implies that structural regularization is not optional but mathematically necessary for compact models to avoid overfitting to noise—the additional capacity required for robustness must be compensated by architectural inductive biases rather than raw parameter count.
Recent advances in complex network theory have provided new perspectives for understanding information propagation in deep neural networks. Liu et al. [
4,
5,
6,
7] developed comprehensive frameworks for analyzing structural coherence and robustness in networked systems under noise disturbance. Their analysis of network topology effects on signal propagation efficiency suggests that architectural modifications optimizing information flow could improve optimization dynamics in deep networks facing noisy inputs.
The Micro Hybrid Attention Module (MHAM) as shown in
Figure 1 is designed for efficiency on edge devices through the overall pipeline. Unlike standard coupled mechanisms, MHAM employs a spectrally decoupled design: the upper branch optimizes channel topology (global context) while the lower branch maintains spatial coherence (local geometry). This parallel processing ensures robust feature learning under noisy label conditions without rank collapse.
While our theoretical framework applies broadly to noisy visual recognition tasks conceptually, in this work we focus our experimental validation on image classification, particularly facial expression recognition (FER) and CIFAR-10/CIFAR-10N benchmarks. Unlike general object classification benchmarks (e.g., CIFAR, ImageNet) where inter-class boundaries are semantically distinct, FER requires capturing the manifold geometry of micro-expressions—subtle facial movements spanning only a few pixels that distinguish emotions like “fear” from “surprise.” Furthermore, FER datasets exhibit inherent subjective noise arising from annotator disagreement on ambiguous expressions, making them ideal testbeds for validating high-frequency noise filtering hypotheses. The combination of (1) weak discriminative signals requiring fine-grained spectral analysis and (2) real-world annotation noise provides a rigorous stress test for our spectral–topological framework that synthetic noise benchmarks cannot replicate.
1.2. Research Objectives
From an algorithmic optimization perspective, this work aims to answer three tightly related research questions about designing robust and efficient lightweight networks under noisy labels.
First, can attention mechanisms be designed and analyzed as structural regularizes that optimize the network’s internal topology for improved gradient flow and loss landscape smoothness?
Second, how does knowledge distillation function as an optimization algorithm in this setting, and what theoretical mechanisms explain its effectiveness in noisy-label scenarios?
Third, what are the fundamental complexity trade-offs in attention mechanism design for lightweight architectures, and how close can practical implementations such as MHAM approach the theoretical efficiency bounds?
1.3. Contributions
The main contributions of this paper are:
We move beyond the conventional “feature refinement” view and propose a spectral graph theoretical framework where MHAM is analyzed as a localized Laplacian smoothing operator, explaining its efficacy in noise suppression through rigorous mathematical formulation.
We analyze the limitations of coupled attention mechanisms (e.g., CBAM), which mix channel and spatial information through standard convolutions and may lead to spectral interference and rank collapse in the feature manifold. In contrast, we propose a strictly decoupled depth-wise design that preserves channel manifold orthogonality and enforces a block-diagonal transformation on the spatial graph, providing theoretical justification for the architectural choices.
We derive closed-form complexity bounds for MHAM and demonstrate its optimality within the class of hybrid channel-spatial attention mechanisms.
We validate the theoretical framework on FER2013 (real-world noisy labels), CIFAR-10 with 40% symmetric synthetic noise, and CIFAR-10N with real human annotation noise, demonstrating improved convergence stability and optimization robustness with state-of-the-art efficiency on the accuracy–parameter Pareto frontier.
1.4. Paper Organization
The remainder of this paper is organized as follows:
Section 2 reviews related work on attention mechanisms, knowledge distillation, and network topology in deep learning.
Section 3 presents the proposed methodology with detailed algorithmic formulations.
Section 4 provides rigorous complexity analysis.
Section 5 presents experimental validation and convergence analysis.
Section 6 discusses theoretical implications and limitations.
Section 7 concludes the paper.
3. Methodology
Notation Convention: Throughout this section, we distinguish between the loss function (calligraphic) and the graph Laplacian (subscripted). Specifically, , , and denote loss functions, while denotes the Laplacian matrix of the spatial graph .
3.1. Problem Formulation
Building upon the residual learning framework [
7], our lightweight backbone incorporates inverted residual blocks to maintain gradient flow. Let
denote a training dataset where
are input images and
are class labels potentially corrupted by noise. We seek to learn a parameterized function
mapping inputs to the probability simplex, where the parameter vector
has minimal cardinality while achieving robust optimization under label uncertainty.
Definition 1 (Optimization Robustness). A learning algorithm is -robust if, over random initializations and data orderings, the variance of final test accuracy is bounded by with probability at least .
Our goal is to design architectural components and training algorithms that maximize optimization robustness while minimizing parameter count and computational complexity.
3.2. Graph Theoretic Formulation of Feature Representations
To rigorously analyze the optimization behavior of the proposed mechanism, we move beyond the standard tensor view and model the intermediate feature representations using Graph Signal Processing (GSP) [
28,
29]. This formulation allows us to quantify “structural coherence” mathematically and provides a principled framework for understanding noise suppression.
Definition 2 (Feature Manifold as a Graph Product). Let
be the feature tensor. We define the underlying structure as a product graph , where
Spatial Graph is a grid lattice representing the spatial domain. Its connectivity is defined by the local receptive field (e.g., or neighborhoods).
Channel Graph represents the inter-dependencies among feature channels.
Definition 3 (Graph Total Variation and Signal Smoothness).
Drawing from GSP theory [28], the “cleanliness” of a feature map is inversely proportional to its Graph Total Variation (GTV). For a specific feature channel (viewed as a graph signal on
), the GTV is defined as the Laplacian quadratic form:Insight: High-frequency noise (e.g., from mislabeled samples or background clutter) manifests as large localized gradients, leading to high values. The optimization goal of MHAM is effectively to minimize relative to while preserving semantic edges that carry discriminative information. Definition 4 (Structural Coherence). The structural coherence of
quantifies the efficiency of information flow through the network. Following [5] and recent work on resistance distance in complex networks [30], high coherence corresponds to:—Concentrated node strength distribution (clear importance hierarchy)—Short effective path lengths (efficient gradient propagation)—High local clustering (preserved local feature relationships)—Minimized resistance distance between semantic clusters. Remark 1 (Consistency with Affective Computing Theory). Our graph-theoretic formulation aligns with recent findings in graph-based emotion recognition. Studies on physiological signal analysis [31,32] demonstrate that emotional states form smooth clusters on appropriately constructed graphs—the Graph Total Variation in genuine emotional responses is significantly lower than that of noise-corrupted signals. By enforcing Laplacian smoothness via MHAM, we implicitly constrain the visual features to respect this inherent topological structure of the “emotion manifold,” making the network robust to label noise which typically violates this smoothness constraint. The optimization objective can be reformulated as finding parameters that maximize structural coherence (minimize GTV) of intermediate representations while minimizing classification loss.
3.3. MHAM: A Spectral Decoupling Optimization Operator
We use the term spectral decoupling in a precise graph Fourier sense: the feature tensor is modeled on the product graph
, where
encodes channel-wise dependencies and
encodes the spatial lattice. A spectrally decoupled operator acts either on
or on
while preserving the eigenbasis of the other factor and avoids mixing signals across the product graph. In this framework, MHAM performs channel-wise topology optimization on
and spatial Laplacian smoothing on
via depth-wise filtering, without introducing cross-channel spectral leakage. We propose the Micro Hybrid Attention Module (MHAM) not merely as a feature refinement block, but as a Spectral Decoupling Operator designed to minimize the Dirichlet energy of noise while preserving semantic signals. Unlike prior works (e.g., CBAM [
9]) that mix channel and spatial information via standard convolution, MHAM relies on a strict theoretical separation of spectral domains.
Definition 5 (MHAM Operator). MHAM is a mapping 𝒜 →defined as:where
is the channel attention map,
is the spatial attention map, and
denotes element-wise multiplication with broadcasting. 3.3.1. Theoretical Justification for Depth-Wise Convolution
Proposition 1 (Spectral Independence). Standard convolution acts as a coupled filter that mixes signals across the combined graph
. In contrast, depth-wise separable convolution acts as a graph filter strictly on , preserving the spectral eigenbasis of individual channels. Analysis: Formally, standard 2D convolution on a C-channel feature map can be viewed as a graph filter on the product graph , with a dense mixing matrix that is not block diagonal with respect to the channel index. In contrast, depth-wise convolution applies a polynomial graph filter independently on each channel, resulting in a block-diagonal operator in the joint spectral basis of .
Let
be the operation of the spatial attention branch. In CBAM,
is a standard
convolution, which implies:
This aggregation across (channels) introduces spectral interference, potentially corrupting the specific topology of one channel with noise from another.
In MHAM, we employ a
depth-wise Convolution. This effectively approximates a polynomial spectral filter
operating independently on the manifold of the aggregated spatial map:
This ensures that the spatial re-weighting map is derived solely from the spatial topology, orthogonal to the channel optimization step.
To rigorously interpret the role of the depth-wise convolution within MHAM from a Graph Signal Processing (GSP) perspective, we introduce the concept of spectral decoupling. In a standard convolution, features are mixed across channels, potentially allowing high-frequency noise from one feature map to propagate to others. Conversely, the depthwise operation in MHAM applies a spatial filter
to each channel
independently:
In the graph spectral domain, this is equivalent to operating on the eigenvectors of the graph Laplacian associated with the c-th channel’s spatial grid, without cross-reference to (where ). This channel-wise independence ensures that noise-induced spectral perturbations are isolated and filtered within their respective channels, preventing “spectral leakage” and thereby enhancing the signal-to-noise ratio of the generated attention maps.
3.3.2. Preservation of Feature Manifold Rank
A critical limitation of coupled attention mechanisms like CBAM (which uses standard convolution for spatial attention) is the risk of rank collapse in the feature manifold. According to Dong et al. [
33], repeated application of mixing matrices (such as
standard convolution) tends to align feature vectors into a lower-dimensional subspace, reducing the discriminative power of the representation. This phenomenon, termed “rank diminution,” is particularly detrimental in lightweight networks where the feature dimensionality is already constrained.
Proposition 2 (Rank Preservation via Block-Diagonal Structure). By employing depth-wise separable convolution, MHAM enforces a block-diagonal structure on the transformation matrix, preserving the rank of the feature manifold.
Analysis: The transformation
operates as:
where
is the local
kernel for channel
. This block-diagonal property ensures that the transformation matrix has rank equal to
, preserving the linear independence of channel vectors.
In contrast, standard coupled attention mechanisms (like CBAM) employ dense transformation matrices through
standard convolution, which can be expressed as:
where the Kronecker product structure inherently reduces the effective rank by coupling all channels through shared spatial weights. According to Dong et al. [
33], such dense mixing forces feature vectors to converge rapidly into a lower-dimensional subspace (rank collapse). By avoiding this dense channel mixing, MHAM maintains a higher-rank feature manifold, ensuring that subtle micro-expression details are not compressed into oblivion during forward propagation.
Theorem 1 (Information Capacity Preservation). Under the block-diagonal constraint, the mutual information between input and output features satisfies:with equality only when the input channels are perfectly correlated. Proof Sketch. The data processing inequality states that post-processing cannot increase mutual information. The block-diagonal structure of preserves channel-wise information independently, while the coupled structure of projects information onto a shared subspace, necessarily losing information about channel-specific variations. This ensures that MHAM maximizes the capacity of the information bottleneck, allowing distinct micro-expression features to remain orthogonal in the high-dimensional space. □
This rank preservation property is particularly crucial for facial expression recognition, where subtle discriminative features (e.g., the distinction between “fear” and “surprise” often lies in minute eyebrow configurations) must be maintained as linearly separable throughout the network depth.
3.3.3. Channel Attention: Vital Feature Identification via Channel Spectra
In our graph-theoretic framework, each feature channel can be regarded as a distinct node in the channel graph , representing a specific visual primitive (e.g., a lip corner curve, an eyebrow ridge). However, not all primitives are equally robust to noise; label corruption often manifests through “distractor channels” that propagate erroneous gradient signals and destabilize optimization.
Drawing parallels to complex network robustness analysis [
4,
5,
6,
7,
8], where system stability critically depends on identifying and protecting a few “vital nodes,” we reinterpret the channel attention mechanism as a vitality selector. Rather than merely re-weighting features, it acts as a learned centrality measure, identifying and amplifying the “hub” channels that carry the most structurally consistent information (low GTV), while attenuating connections to noisy channels (high GTV). The detailed computational procedure is summarized in Algorithm 1.
| Algorithm 1: Channel Attention Computation |
| Input: Feature tensor . |
| Output: Channel attention . |
| 1. . (Shape: ) |
| 2. . (Shape: ) |
| 3. ; . |
| 4. . |
| 5. Return . |
The mathematical formulation is:
This step re-weights the graph signals to suppress channels dominated by noise (i.e., channels with high random entropy).
Proposition 3 (Channel Attention as Node Importance Weighting). The channel attention map assigns importance weights to channel nodes in based on aggregated statistics. Channels with higher average and maximum activations receive larger weights, effectively pruning low-importance channels from the information flow graph.
Proof Sketch. The sigmoid-activated MLP output is bounded in . Multiplication by scales each channel’s contribution to downstream computations. Channels with near-zero attention weights contribute minimally to gradient flow during backpropagation, equivalent to soft pruning in the graph topology. □
Remark 2 (Connection to Vital Node Theory). From the perspective of complex network analysis, the channel attention mechanism performs Vital Node Identification on the channel graph [32]. In network science, vital nodes are defined as vertices whose removal causes maximal disruption to network function—equivalently, nodes through which the majority of information flows. The learned attention weights can be interpreted as a differentiable approximation to node importance metrics such as betweenness centrality or eigenvector centrality:where is the number of shortest paths from node to , and is the number passing through channel . By learning to identify and amplify vital channels while suppressing peripheral ones, MHAM concentrates gradient flow through the most informative pathways, effectively implementing an adaptive network pruning strategy that preserves the network’s information-theoretic capacity while reducing vulnerability to noise propagation through non-vital channels. 3.3.4. Spatial Attention: Laplacian Coherence Filtering
The spatial attention branch optimizes coherence across the 2D spatial lattice by identifying discriminative spatial locations. From the perspective of Graph Neural Networks (GNNs), the
depth-wise convolution aggregates 2-hop neighbor information on the grid graph, acting as a low-pass filter that smooths the attention map. The specific computational steps are outlined in Algorithm 2.
| Algorithm 2: Spatial Attention Computation |
| Input: Channel-refined tensor . |
| Output: Spatial attention . |
| 1. . (Shape: ) |
| 2. . (Shape: ) |
| 3. Concatenate: . (Shape: ) |
| 4. Apply depthwise separable convolution: . (Shape: ) |
| 5. Return . |
The mathematical formulation is:
Proposition 4 (Spatial Attention as Path Length Reduction). The spatial attention map effectively shortens path lengths in
by amplifying spatially coherent regions and suppressing isolated noise.
Proof Sketch. The 5 × 5 convolution integrates local neighborhood information, assigning higher weights to spatially coherent regions where neighboring nodes have similar activations. This is equivalent to preferentially routing information through dense local clusters while bypassing isolated noise nodes. □
Theorem 2 (Laplacian Regularization). The multiplicative application of acts as a Tikhonov regularizer on the feature graph. If
highlights regions of high structural coherence, the operation
effectively modifies the gradient flow
to penalize high-frequency variations (noise) in the background while preserving low-frequency components (semantic shapes) in the foreground.
Proof Sketch. The gradient of the loss with respect to input features involves the term
. Since
is generated by a low-pass
filter, it suppresses high-frequency gradients corresponding to isolated pixel noise. Thus, the update rule approximates Laplacian Smoothing:
, where
denotes the spatial graph Laplacian. This promotes the formation of coherent feature clusters—a property explicitly sought in the analysis of robust complex networks [
4,
5,
6]. □
Remark 3 (Spectral Low-Pass Filter Interpretation). While implemented as a standard depth-wise convolution, MHAM’s behavior can be analyzed through the lens of spectral graph filtering [28] to explain its effectiveness. For any graph signal on the spatial lattice , convolution applies a filter function to the eigenvalues of the graph Laplacian :where contains the eigenvectors of . Our Depth-wise Kernel can be understood as approximating a heat kernel , which attenuates high eigenvalues () corresponding to high-frequency noise while preserving low eigenvalues () corresponding to smooth semantic structures. This analysis provides a principled explanation for why the specific kernel size offers a superior trade-off—not merely due to receptive field size, but because of the resulting spectral smoothing properties that align with Low-Pass Filtering (LPF) principles in graph signal denoising [29]. 3.3.5. Sequential Composition and Gradient Flow
The sequential application of channel and spatial attention modifies the backward gradient flow:
The multiplicative terms
rescale gradients, with larger values for attended (important) features and smaller values for suppressed features. This creates an adaptive learning rate effect that accelerates convergence on discriminative patterns. The complete forward pass of the MHAM module is detailed in Algorithm 3.
| Algorithm 3: MHAM Forward Pass |
| Input: Feature map . |
| Output: Refined feature map . |
| 1. Channel context (global branch): |
| i. . (Shape: ) |
| ii. . (Shape: ) |
| 2. Spatial coherence (local branch; depth-wise filtering): |
| i. . (Shape: ) |
| 3. Fusion and gating: |
| i. Broadcast to . |
| ii. . (Shape: ) |
| 4. Refinement: . |
| 5. Return . |
3.3.6. Comparison with CBAM and Theoretical Justification
Unlike CBAM [
9], which employs a large-kernel (
) standard convolution for spatial attention, MHAM utilizes a
depth-wise separable convolution. This design choice is not arbitrary but theoretically grounded in the principle of feature orthogonality.
Standard convolution in CBAM implicitly mixes channel information during spatial recalibration, potentially interfering with the channel importance learned in the previous step. By using depth-wise convolution, MHAM ensures that spatial coherence is optimized within each channel manifold independently, preserving the topology established by the channel attention.
Furthermore, the kernel size provides a receptive field that aligns better with the local coherence length of micro-expressions (e.g., local edge shifts) compared to the broader context, effectively filtering high-frequency noise while retaining structural edges.
The theoretical advantage of this decoupling can be understood through the lens of optimization stability: by preventing cross-domain interference, MHAM reduces the condition number of the effective Hessian, leading to smoother loss landscapes and more stable convergence trajectories. A comparison of the architectural specifications and theoretical properties is summarized in
Table 1.
3.3.7. On the Optimality of Kernel Size and Coherence Length
The choice of a kernel in MHAM is not empirical but relates to the Characteristic Coherence Length of facial micro-expressions. This concept, borrowed from signal processing and network theory, refers to the spatial scale over which features maintain statistical correlation.
Definition 6 (Coherence Length). For a feature map , the coherence length is defined as the characteristic decay length of the spatial autocorrelation function:In a standard feature map (typical for mobile architectures at intermediate layers), facial expression primitives—such as the curvature of a lip corner, the tension of nasolabial folds, or the subtle raise of an eyebrow—span approximately 4–6 pixels in spatial extent. Proposition 5 (Matched Filter Optimality). The
kernel acts as an optimal matched filter for mesoscopic facial features, maximizing the signal-to-noise ratio (SNR) for discriminative expression patterns.
Analysis:
- −
A 3 × 3 kernel (1-hop neighborhood) is too narrow to capture the full topological curvature of expression primitives. It functions primarily as an edge detector, missing the spatial context necessary to distinguish between similar edge configurations (e.g., a smile versus a grimace both involve curved edges but with different spatial extents).
- −
A 7 × 7 kernel (3-hop neighborhood) introduces excessive background context into the receptive field. In the presence of label noise, this larger context dilutes the discriminative signal with irrelevant background variations, effectively lowering the SNR.—The 5 × 5 kernel (2-hop neighborhood) provides the Goldilocks zone: sufficient context to capture the coherent structure of expression primitives while excluding distant noise sources.
This analysis aligns with the “Small-World” property in network theory [
4,
5,
6,
7], where optimal information propagation requires balancing local clustering (captured by the
receptive field) with path efficiency (achieved by the depth-wise structure that avoids unnecessary cross-channel routing).
3.4. Knowledge Distillation: An Entropy-Constrained Interpretation
Rather than proposing a novel loss function, we provide a thermodynamic interpretation of the standard knowledge distillation (KD) objective that explains its effectiveness in noisy label scenarios. Although the formulation aligns with classical KL-divergence minimization [
15], our contribution lies in identifying its role as an implicit entropy constraint mechanism during training under label noise.
Definition 7 (Entropy-Constrained Distillation).
Given a teacher network with output distribution and student network with output , the standard distillation objective is: We observe that this objective implicitly enforces an entropy constraint: .
In the presence of label noise, corrupted annotations typically manifest as low-entropy (one-hot) distributions that encourage the model to overfit to incorrect classes with high confidence. By enforcing consistency with the high-temperature teacher distribution, the distillation objective functions as an entropy-constrained regularizer. It prevents the student’s output distribution from collapsing into the sharp, deterministic spikes associated with noisy labels, thereby forcing the model to learn smoother, multi-modal features that are statistically more consistent with the true underlying data manifold.
Theorem 3 (Variance Reduction Property). Under mild regularity conditions, training with soft targets reduces the variance of gradient estimates compared to hard targets: Proof Sketch. The soft target spreads probability mass across classes, reducing the magnitude of per-sample gradient contributions from any single class. For a sample with true label , the hard label gradient has magnitude proportional to , while the soft label gradient is distributed across classes according to . By Jensen’s inequality applied to the squared gradient norm, the variance reduction follows. □
Theorem 4 (Geometric Regularization). The KL divergence term constrains the student’s logit geometry to match the teacher’s inter-class relationships: Proof. At the KL minimum, , implying for some constant . Thus logit differences are preserved: . □
This geometric constraint transfers the teacher’s learned class structure to the student, providing regularization against label noise by anchoring predictions to a cleaner distribution.
3.5. Network Architecture Integration
The complete MHAM-Net architecture integrates MHAM modules into a MobileNetV3-Small backbone at strategic positions corresponding to different feature resolutions.
Definition 8 (Multi-Scale Topology Optimization). MHAM modules are inserted after bottleneck blocks 3, 6, and 9, operating at spatial resolutions of 1/4, 1/8, and 1/16 of the input, respectively. This multi-scale placement optimizes coherence at different levels of the feature hierarchy.
The forward pass is:
where
denotes the
-th bottleneck block and
denotes the MHAM operator at scale
.
5. Experimental Validation
5.1. Experimental Setup
Datasets: We evaluate on FER2013, containing 28,709 training images, 3589 validation images, and 3589 test images across seven expression categories. FER2013 is characterized by significant label noise with approximately 30% annotation ambiguity [
24]. We follow the official FER2013 split provided by the original challenge, without any subject-level filtering, to facilitate comparison with prior work. In addition, we evaluate on CIFAR-10 and CIFAR-10N. CIFAR-10 contains 50,000 training and 10,000 test images of size 32 × 32 across 10 object categories. CIFAR-10N provides human-annotated noisy labels for the CIFAR-10 training set under five protocols (Aggregate, Random1/2/3, and Worst), representing realistic asymmetric and instance-dependent label noise.
Implementation Details:
Backbone: MobileNetV3-Small (width_mult = 0.5); MHAM configuration: r = 16, K = 5, inserted at blocks 3, 6, 9; Teacher: ResNet-50 pretrained on ImageNet, fine-tuned on FER2013; Distillation: τ = 6, α = 0.3; Optimizer: Adam, initial learning rate 0.001, cosine annealing; Training: 100 epochs, batch size 64, early stopping on validation loss; Reproducibility: Five independent runs with different random seeds.
Input preprocessing: FER2013 images are 48 × 48 grayscale; we replicate the single channel to obtain 3-channel inputs and resize to 112 × 112 using bilinear interpolation. All images are normalized to (0, 1) and standardized using ImageNet mean and variance. CIFAR-10 and CIFAR-10N images are kept at 32 × 32 and normalized using the standard CIFAR-10 statistics.
Data augmentation: For all datasets, we apply random horizontal flipping with probability 0.5 and random cropping with 4-pixel padding. No additional advanced augmentations (e.g., Cutout, Mixup) are used to keep the training pipeline lightweight.
Environment: Experiments are conducted on an NVIDIA GPU (RTX 5070Ti) with 16 GB memory, Intel CPU 265K, and 96 GB RAM, using PyTorch 2.7 and CUDA 12.7.
A complete list of training hyperparameters for FER2013, CIFAR-10, and CIFAR-10N (optimizer, learning rate schedule, batch size, early stopping criteria, and distillation settings) is summarized in
Table A1 in the
Appendix A.
The source code and trained models will be released upon acceptance to facilitate replication and adoption in the community.
5.2. Convergence Analysis
Figure 2 presents the training dynamics comparing baseline MobileNetV3-Small with MHAM-Net under knowledge distillation. As shown in
Figure 2a, the baseline MobileNetV3-Small exhibits high variance and oscillatory convergence, whereas
Figure 2b shows that MHAM-Net with KD converges smoothly with reduced inter-run variance.
As shown in
Table 4, the coefficient of variation (CV) reduction from 3.44% to 2.63% represents a 23% improvement in optimization robustness, validating Theorem 2’s variance reduction prediction.
5.3. Loss Landscape Analysis
To quantitatively assess the impact of MHAM and KD on the loss landscape, we compute standard sharpness-related metrics following prior work, including the trace of the Hessian, an approximate basin width, and the train–test generalization gap. The results in
Table 5 compare the baseline, +MHAM, and +MHAM + KD configurations on FER2013.
The 53% reduction in sharpness and 122% increase in basin width provide empirical support for Propositions 1 and 3 regarding MHAM’s regularization effects.
Remark 4 (Connection to Sharpness-Aware Minimization). This observation aligns with the principle of Sharpness-Aware Minimization [6], which posits that flatter minima result in lower generalization error bounds, especially in the presence of label noise. SAM explicitly seeks flat minima by minimizing the worst-case loss within a neighborhood: Our framework achieves an analogous effect through an implicit sharpness-aware mechanism. By filtering out high-frequency spectral noise in the feature maps (Remark 3), MHAM prevents the network from overfitting to sharp, high-curvature directions in the loss landscape that are typically associated with label noise. The spectral low-pass filtering effectively constrains the optimization trajectory to remain within flat basins, achieving the benefits of SAM without explicit adversarial perturbations. This provides a novel architectural pathway to flat-minima generalization, complementing algorithmic approaches like SAM.
Furthermore, the topology-aware regularization from MHAM aligns with recent findings on Resistance Distance Dominant Graph Neural Networks (RDDGN) [
35], which demonstrate that networks respecting the intrinsic geometric structure of data graphs exhibit improved robustness. The GTV minimization objective (Definition 3) implicitly minimizes the effective resistance distance between semantically similar features, promoting the formation of robust feature clusters that generalize well under distribution shift.
5.4. Performance on Real-World Noisy-Labels
To verify the effectiveness of the proposed method in practical application scenarios, we extended our evaluation to the CIFAR-10N dataset. Unlike synthetic datasets where noise is generated mathematically (e.g., symmetric flipping), CIFAR-10N contains real-world human annotation errors collected from Amazon Mechanical Turk. This dataset introduces asymmetric and instance-dependent noise, which poses a significantly greater challenge for lightweight models.
Table 6 presents the performance of the proposed MHAM+KD framework across different noise configurations defined in CIFAR-10N. The ‘Aggregate’ set represents consensus labels with a low noise rate (~9%), while ‘Random 1–3’ represents labels from individual annotators. Notably, the ‘Worst’ set contains extreme noise (~40.21%) derived from the poorest-performing annotators.
The experimental results demonstrate the robustness of the proposed method. Firstly, the method maintains highly consistent performance across different individual annotators (Random 1, 2, and 3), with accuracy fluctuating by less than 0.5%. This indicates that MHAM is insensitive to the specific subjective bias of different human labelers. In the scenario, where over 40% of the training labels are incorrect, the proposed method prevents model collapse, achieving a test accuracy of 63.93%. This suggests that the spectral filtering nature of MHAM effectively suppresses the gradients from noisy samples, allowing the model to learn dominant features even when the supervision signal is heavily corrupted. On the cleaner ‘Aggregate’ labels, the model achieves 71.86%, confirming that the noise-tolerance mechanism does not compromise the model’s capacity to learn from clean data.
We further validate our method on CIFAR-10N, which provides real-world human-annotated noisy-labels for CIFAR-10 under five protocols: Aggregate, Random1, Random2, Random3, and Worst. These settings correspond to different annotator groups and noise severities and represent asymmetric and instance-dependent noise that is more challenging than synthetic symmetric corruption.
As shown in
Table 6, across all real-noise protocols, MHAM+KD achieves consistent accuracy in the range of 63.9–71.9%, demonstrating robustness to asymmetric and instance-dependent label noise. The performance degradation from Aggregate to Worst noise remains moderate, indicating that the spectral–topological filtering of MHAM and the entropy-constrained KD objective transfer well beyond synthetic noise settings.
5.5. Comparison with Alternative Attention Modules
To demonstrate that the performance improvement stems from the specific design of MHAM rather than simply increasing model capacity, we compared the proposed module against widely used lightweight attention mechanisms: Squeeze-and-Excitation (SE), Efficient Channel Attention (ECA), and the Convolutional Block Attention Module (CBAM). All attention modules were integrated into the same MobileNetV3 backbone and trained under identical conditions using CIFAR-10 with 20% synthetic symmetric noise.
Table 7 summarizes the results averaged over five independent runs. The proposed MHAM achieves the highest mean accuracy of 70.00%. While SE and CBAM also provide robustness improvements over the baseline (MobileNetV3 without attention), MHAM demonstrates superior stability with the lowest standard deviation (0.009). Theoretically, this advantage can be attributed to the specific frequency-domain filtering characteristics of MHAM, which are more effective at isolating structural features from label-noise-induced gradients compared to the global average pooling strategies primarily used in SE and CBAM.
Compared to CBAM, MHAM slightly improves accuracy by 0.47 percentage points under a similar parameter and FLOP budget, demonstrating that the decoupled depth-wise design does not sacrifice performance despite being theoretically better conditioned.
5.6. Ablation Studies
We conducted ablation studies on FER2013 to disentangle the contributions of channel attention, spatial attention, and knowledge distillation.
Table 8 reports accuracy, parameter count, FLOPs, and convergence speed for different combinations of these components.
The ablation reveals: 1. Channel and spatial attention provide complementary benefits. 2. Their combination (MHAM) yields super-additive improvements in convergence speed. 3. Knowledge distillation and MHAM are synergistic, with combined gains exceeding individual contributions.
5.7. Sensitivity to Label Noise
To investigate how MHAM+KD behaves under different levels of label corruption, we injected varying amounts of symmetric noise into FER2013 and reported the resulting performance.
Table 9 summarizes the baseline and MHAM+KD accuracies across noise rates from 0% to 40%.
The increasing relative improvement at higher noise levels validates the theoretical prediction that entropy-constrained distillation provides stronger regularization benefits under noisier conditions.
5.8. Comparison with State-of-the-Art
Under an identical MobileNetV3-Small backbone and similar parameter/FLOP budgets, MHAM achieves the best mean accuracy (69.98%) with competitive variance (0.93%), slightly outperforming SE (69.67%) and CBAM (69.51%). Compared to ECA and the no-attention baseline, MHAM yields large absolute gains of more than 10 percentage points, confirming that the proposed decoupled design is at least as effective as standard coupled attention while preserving the theoretical benefits analyzed in
Section 3.3. For context,
Table 10 lists the performance of heavyweight reference models, although they are not the primary focus of this lightweight comparison.
As summarized in
Table 11 and visualized in
Figure 3, MHAM-Net lies on the accuracy–parameter Pareto frontier in the ultra-lightweight regime.
Pareto Efficiency Analysis: Our contribution lies in the efficiency-accuracy trade-off, not absolute accuracy. While PAtt-Lite achieves marginally higher accuracy (+1.3%), it requires 3.6× more parameters (1.9M vs. 0.52M). MHAM-Net achieves a 3.6× improvement in parameter efficiency ratio (131.7 vs. 36.7), establishing a new point on the Pareto frontier for ultra-lightweight FER models.
Among ultra-lightweight models (<1M parameters), MHAM-Net achieves competitive accuracy with the smallest parameter count and computational cost while incorporating principled noise handling through topological regularization.
5.9. Validation on CIFAR-10 with Synthetic Label Noise
To further validate the generalizability of our framework beyond real-world noisy FER datasets, we conducted comprehensive experiments on CIFAR-10 with controlled synthetic label noise. This enables precise quantification of robustness under known noise conditions.
5.9.1. Dataset and Implementation Details
Dataset: We evaluated on CIFAR-10, consisting of 50,000 training images and 10,000 test images across 10 classes. To simulate challenging noisy environments, we injected 40% symmetric label noise into the training set by flipping labels to other classes with uniform probability, while keeping the test set clean for unbiased evaluation.
Experimental Setup: (a) Student Model: MobileNetV3-Small (Baseline) vs. MobileNetV3-Small with MHAM (Proposed Method); (b) Teacher Model: ResNet-50 pre-trained on the noisy CIFAR-10 training set; (c) Optimizer: AdamW with initial learning rate 0.001 (reduced from 0.005 to ensure stable convergence); (d) Learning Rate Schedule: Cosine Annealing over 100 epochs; (e) Batch Size: 128; (f) Distillation Settings: Temperature T = 4.0, distillation weight α = 0.3; (f) MHAM Initialization: Identity-like initialization to prevent signal damping in early training; (g) Reproducibility: Random seed fixed to 42; experiments conducted on NVIDIA GPU using PyTorch. (h) Evaluation protocol: For FER2013, we selected the model with the best validation accuracy and reported the corresponding test accuracy. For CIFAR-10 and CIFAR-10N, we evaluated on the clean test set at the end of training. All reported numbers are averaged over 5 random seeds, and we provide mean ± standard deviation where applicable.
The learning rate reduction from 0.005 to 0.001 was critical for stable convergence. Higher learning rates caused oscillation and divergence, particularly under high noise conditions. The identity-like initialization for MHAM ensured that early-stage feature extraction was not impeded by random attention weights.
5.9.2. Main Results
As shown in
Table 12, the standard MobileNetV3-Small baseline achieves only 54.85% peak accuracy under 40% label noise. Due to its lightweight capacity (~2.5M parameters), the baseline struggles to differentiate between clean and noisy samples, leading to significant performance degradation. In contrast, our proposed method achieves 60.10% accuracy, representing an absolute improvement of +5.25 percentage points.
This improvement is particularly significant given the lightweight architecture constraints. The 5.25% gain demonstrates that: 1. MHAM effectively identifies task-relevant features even when 40% of training signals are corrupted. 2. Soft teacher guidance prevents overfitting to hard (potentially incorrect) labels. 3. The synergy between attention and distillation provides robust regularization under extreme noise conditions.
5.9.3. Convergence and Stability Analysis
Validation accuracy curves comparing baseline (gray dashed) and proposed method (red solid) on CIFAR-10 with 40% symmetric noise over the 100 training epochs is demonstrated in
Figure 4. The baseline exhibits instability in later epochs, while our method shows smooth, monotonic improvement.
As shown in
Figure 4, the baseline model (gray dashed line) exhibits significant oscillation, particularly in later training stages (epochs 60–100). This fluctuation indicates the model is oscillating between learning true patterns and memorizing noisy-labels—a hallmark of overfitting to label noise. Our approach (red solid line) demonstrates remarkably smooth convergence. As the learning rate decays via Cosine Annealing, the model shows steady, continuous improvement without the oscillatory behavior of the baseline. Our method surpasses the baseline as early as epoch 10 and maintains a consistent lead throughout training, indicating that MHAM enables faster extraction of discriminative features even under noisy supervision. The smooth trajectory suggests that MHAM acts as an effective feature-level noise filter, stabilizing gradient updates even when input labels are unreliable. This aligns with our theoretical analysis of MHAM as a spectral low-pass filter (Remark 3).
5.9.4. Uncertainty Analysis via Prediction Entropy
Beyond accuracy, we investigated prediction confidence by monitoring the Shannon Entropy of output probability distributions. As shown in
Table 13, our method reduces the converged prediction entropy from 1.972 to 1.894. Lower entropy indicates higher model confidence and sharper decision boundaries.
The baseline model maintains high entropy (1.972) at convergence, as illustrated in
Figure 5, reflecting a state of “confusion” caused by conflicting noisy labels during training. The model has partially memorized noise, leading to uncertain predictions even on clean test data.
In contrast, our method successfully reduces entropy to 1.894—a reduction of 0.078 (approximately 4% relative improvement). This reduction has important implications:
Sharper Decision Boundaries: Lower entropy indicates the model has learned more decisive classification boundaries, assigning higher probability mass to the predicted class.
Reduced Memorization: By aligning with the teacher’s soft targets rather than hard (noisy) labels, and by focusing on salient features via MHAM, the student network avoids memorizing noise patterns.
Manifold Learning: The entropy reduction suggests our model successfully learns a low-dimensional manifold of the clean data distribution, filtering out the high-dimensional noise perturbations that would otherwise inflate prediction uncertainty.
Entropy analysis provides direct empirical validation of our core hypothesis, that the combination of spectral attention filtering (MHAM) and entropy-constrained distillation (KD) helps lightweight networks establish robust decision boundaries under severe label noise. Importantly, this ~4% relative reduction in prediction entropy is achieved with less than 1% additional FLOPs per MHAM module (see
Section 4.1), indicating that the uncertainty reduction comes at negligible computational cost.
6. Discussion
6.1. Theoretical Implications
The experimental results across both FER2013 and CIFAR-10 benchmarks support our theoretical framework in several ways:
First, the improved convergence and flatter loss landscapes observed with MHAM align with the prediction that attention acts as a coherence filter on the feature lattice graph. The 53% reduction in Hessian trace (FER2013) and smooth convergence curves (CIFAR-10) suggest that multiplicative gating indeed smooths the optimization surface.
Secondly, the consistent variance reduction (23% lower CV on FER2013) and improved performance under label noise validate Theorem 2’s prediction. The entropy analysis on CIFAR-10 provides direct evidence: prediction entropy decreased from 1.972 to 1.894, confirming that entropy-constrained distillation helps establish sharper decision boundaries. This entropy-constrained view of distillation provides a principled explanation for its effectiveness beyond simple knowledge transfer.
Thirdly, the super-additive gains from combining MHAM and KD suggest they address complementary aspects of the optimization challenge—MHAM regularizes the feature space geometry while KD regularizes the output space geometry. The +5.25% improvement on CIFAR-10 with 40% noise demonstrates this synergy under extreme conditions.
6.2. Limitations and Future Work
While our framework demonstrates robust performance on the real-world noisy FER2013 dataset, we acknowledge certain limitations that provide directions for future research. Therefore, our empirical claims are currently limited to lightweight image classification settings rather than covering all visual recognition tasks.
Our experimental validation focuses on the efficiency-constrained regime (mobile FER) with a single dataset. While we have additionally validated our framework on CIFAR-10N with real-world human annotation noise, further benchmarking on larger-scale noisy-label datasets such as Clothing1M and on non-classification tasks (e.g., detection or segmentation) remains as future work. However, we emphasize that FER2013 represents real-world noisy data with inherent annotation ambiguity arising from subjective human perception, which is arguably more challenging and practically meaningful than synthetic noise benchmarks where label corruption follows known statistical patterns.
The comparison with sophisticated noisy label methods (like Co-teaching and DivideMix) is indirect, as our primary focus is on architectural regularization under strict parameter budgets (<1M parameters) rather than complex training pipelines. These methods require 40× more parameters and employ computationally expensive dual-network training or semi-supervised learning schemes. Our contribution lies in achieving comparable noise robustness through efficient architectural design rather than training complexity.
While we provide spectral graph theoretical interpretations of MHAM, formal proofs establishing precise relationships between attention operations and graph-theoretic coherence measures remain for future work. The current analysis provides intuitive justifications that are empirically validated but would benefit from more rigorous mathematical treatment.
The complexity analysis assumes standard CNN backbones; extension to transformer architectures and vision transformers (ViT) requires separate analysis and may yield different efficiency-accuracy trade-offs.
Our theoretical connections to complex network theory draw primarily from recent work on network coherence and robustness [
4,
5,
6,
7,
8]. While these references represent focused work from a single research group, they provide the most comprehensive and mathematically rigorous treatment of coherence dynamics in networked systems currently available. Future work should explore connections to the broader network science literature, including alternative formulations of vital node identification [
41] and resistance distance analysis [
30], to strengthen the theoretical foundations across diverse methodological perspectives.
6.3. Future Directions
Our future research aims to deepen the theoretical and architectural foundations of this framework. We plan to establish formal coherence metrics and derive PAC-Bayes generalization bounds to explicitly quantify the link between attention mechanisms and model generalization. Furthermore, we intend to extend MHAM into a differentiable topology learning framework, enabling the adaptive discovery of optimal attention patterns beyond fixed configurations.
7. Conclusions
Our work presents a complexity-aware optimization framework for lightweight neural networks operating under noisy label conditions. We introduced the Micro Hybrid Attention Module (MHAM) and provided a novel spectral graph theoretical interpretation, demonstrating that MHAM functions as a spectral decoupling operator that separates channel-wise topology optimization from spatial coherence filtering. By reformulating knowledge distillation as entropy-constrained optimization, we further enhanced the model’s geometric regularization properties.
Experimental validation confirmed our theoretical predictions across two complementary benchmarks. On FER2013 (real-world noise), MHAM-Net achieved 68.5% accuracy with only 0.52M parameters, establishing a new state-of-the-art on the efficiency-accuracy Pareto frontier. While methods like PAtt-Lite achieve marginally higher accuracy (+1.3%), they require 3.6× more parameters, highlighting the exceptional efficiency of our approach. On CIFAR-10 (40% synthetic noise), our method achieved 60.10% accuracy, outperforming the MobileNetV3 baseline (54.85%) by +5.25%, while significantly reducing prediction entropy (1.97 to 1.89). On CIFAR-10N, MHAM-Net with KD consistently outperforms the MobileNetV3 baseline under multiple real-noise protocols, further confirming the robustness of our design in realistic noisy-label scenarios. These results provide direct evidence that our framework creates sharper decision boundaries and effectively suppresses noise memorization.
Theoretically, this work establishes a principled connection between attention mechanisms and optimization stability. We suggest that robust deep learning under noise is fundamentally a problem of network topology optimization—designing architectures that concentrate information flow through vital pathways while filtering high-frequency perturbations. This perspective moves beyond empirical architecture search, offering a mathematically grounded path for designing robust, ultra-lightweight networks.