ADMC: Attention-based Diffusion model for Missing modalities Completion


Abstract

Multimodal emotion and intent recognition is essential for automated human-computer interaction, and it aims to analyze users’ speech, text, and visual information to predict their emotions or intent. One of the significant challenges is that missing modalities due to sensor malfunctions or incomplete data. Traditional methods that attempt to reconstruct missing information often suffer from over-coupling and imprecise generation processes, leading to suboptimal outcomes. To address these issues, we introduce an Attention-based Diffusion model for Missing Modalities feature Completion (ADMC). Our framework independently trains feature extraction networks for each modality, preserving their unique characteristics and avoiding over-coupling. The Attention-based Diffusion Network (ADN) generates missing modality features that closely align with authentic multimodal distribution, enhancing performance across all missing-modality scenarios. Moreover, ADN’s cross-modal generation offers improved recognition even in full-modality contexts. Our approach achieves state-of-the-art results on the IEMOCAP and MIntRec datasets, with up to 9.4% improvement in Average Weighted Accuracy, demonstrating its effectiveness.

<ccs2012> <concept> <concept_id>00000000.0000000.0000000</concept_id> <concept_desc>Do Not Use This Code, Generate the Correct Terms for Your Paper</concept_desc> <concept_significance>500</concept_significance> </concept> <concept> <concept_id>00000000.00000000.00000000</concept_id> <concept_desc>Do Not Use This Code, Generate the Correct Terms for Your Paper</concept_desc> <concept_significance>300</concept_significance> </concept> <concept> <concept_id>00000000.00000000.00000000</concept_id> <concept_desc>Do Not Use This Code, Generate the Correct Terms for Your Paper</concept_desc> <concept_significance>100</concept_significance> </concept> <concept> <concept_id>00000000.00000000.00000000</concept_id> <concept_desc>Do Not Use This Code, Generate the Correct Terms for Your Paper</concept_desc> <concept_significance>100</concept_significance> </concept> </ccs2012>

1 Introduction↩︎

Multimodal emotion and intent recognition [1][3] is pivotal for automated human-computer interaction, as it seeks to interpret and identify human emotions and intentions through various modalities such as speech content, voice tones, and facial expressions. However, real-world applications often face the challenge of missing modalities due to incomplete data collection or sensor malfunctions, including speech recognition errors, user silence, or visual occlusion, as shown in Fig. 1. These issues significantly complicate multimodal recognition.

To tackle missing modalities, several existing studies [4][9] focus on leveraging available modalities to reconstruct the missing information, thus mitigating the adverse effects on model performance caused by modality loss. Approaches like those in [6], [10] utilize modality translation to create a unified representation, but this requires separate reconstruction models for each missing modality scenario, leading to increased costs and inconvenience. To address this, [7] proposed a unified framework capable of handling all missing modalities within a single model. This framework, known as the Missing Modality Imagination Network (MMIN), employs cascade residual auto-encoders [5] and cycle consistency learning to generate the missing modalities. Building upon MMIN, [11] enhances model performance by introducing modality-invariant features to bridge modality gaps.

Figure 1: In real-world human-computer interaction, modality missingness frequently occurs due to factors such as speech recognition errors, user silence, or visual occlusion. These issues significantly impact the performance of multimodal recognition and pose challenges to the effective fusion of multimodal information.

Despite advancements, significant challenges persist. Current methods often attempt to recover the multimodal joint representation by jointly training feature extraction and classification networks. However, this approach can lead to over-coupling, causing a sharp decline in performance when modalities are missing. Even with completed modalities, the performance may fall short compared to simple zero padding, which is unsatisfactory. Moreover, these methods often lack a precise generation process, making it difficult to produce missing modality features that closely align with authentic multimodal feature distribution, leading to minimal model improvements.

In this paper, we propose an Attention-based Diffusion model for Missing Modalities feature Completion framework (ADMC), as illustrated in Fig. 2. To prevent over-coupling between features and the classification network, we independently train feature extraction networks for each modality through single-modality classification, preserving their unique characteristics. Subsequently, we introduce an Attention-based Diffusion Network (ADN) to generate missing modality features that closely adhere to the authentic multimodal feature distribution. ADN operates in the latent space of full-modality features, training a denoising autoencoder with a self-attention mechanism to capture inter-modal dependencies. In the presence of missing modalities, the module completes them through a sequence of denoising steps. Finally, to ensure optimal performance of available modalities even when some are missing, we train separate multimodal feature fusion networks (MF) for each missing modality scenario. MF employs a self-attention mechanism to calculate the weights for each modality and aggregates them accordingly. Interestingly, we find that even in full-modality scenarios, using ADN for cross-modal generation can provide supplemental information, enhancing the final recognition. Thus, our method excels in both missing and full-modality scenarios, achieving state-of-the-art results on the IEMOCAP and MIntRec benchmarks.

Our contributions can be summarized as follows:

  • We introduce a novel ADMC framework for missing modality completion that integrates independently trained feature extraction networks and an attention-based diffusion model.

  • We propose an Attention-based Diffusion Network (ADN) for capturing inter-modal dependencies and generating accurate missing modality features.

  • Our method excels in both missing and full-modality scenarios, showcasing strong practical applicability.

  • We evaluated our approach on the IEMOCAP and MIntRec datasets, achieving state-of-the-art results.

2 RELATED WORK↩︎

2.1 Incomplete Multimodal Learning↩︎

To address the issue of missing modalities, simple strategies like filling missing modality values with zeros [12] or averages [13]. However, these methods lack supervision information, leading to suboptimal performance. Low-rank imputation methods attempt to recover missing data by restoring the low-rank nature of complete multimodal data. For instance, Fan et al. [14] minimized the tensor tubal rank to handle various missing modalities. Liang et al. [15] incorporated non-linear functions to capture complex correlations in tensor rank minimization. However, these methods may overlook the complementarity between heterogeneous modalities, resulting in suboptimal results.

Given the powerful feature representation capabilities of deep learning methods, they have shown remarkable performance in reconstructing missing modalities. Early approaches include Autoencoders (AE) [16], Generative Adversarial Networks (GAN) [17], and Variational Autoencoders (VAE) [18]. Recently, Tran et al. [5] proposed a Cascade Residual Autoencoder (CRA) to reconstruct missing modality data. Zuo et al. [11] and Zhao et al. [7] employed CRA and cycle consistency learning to learn joint multimodal representations for addressing missing modality problems, gaining significant attention due to their superior performance. Yuan et al. [19] and zeng et al. [8] utilized Transformer architectures to integrate cross-modal information and capture inter-modal dependencies, achieving outstanding results in missing modality completion tasks. Additionally, Lian et al. [20] introduced Graph Neural Networks (GNN) to capture speaker and temporal dependencies, effectively enabling the reconstruction of missing modalities. These methods have significantly advanced research in the field of incomplete multimodal learning.

2.2 Diffusion Model↩︎

Diffusion models were first applied in the field of image processing. Ho et al. [21] significantly improved the quality of generated samples by simplifying the training objective (directly predicting noise) and optimizing model parameterization. Compared to other generative models, such as GANs and VAEs, these improvements demonstrated a more stable training process. Building on this foundation, Song et al. [22] proposed a unified framework for diffusion models by integrating score matching with stochastic differential equations (SDE), addressing the inefficiency of traditional diffusion models. These advancements have established a solid theoretical foundation and practical methods for high-efficiency diffusion models.

With the further development of diffusion models, their conditional generation capabilities have been progressively explored and optimized. Dhariwal et al. [23] and Ho et al. [24] leveraged the gradients of pre-trained classifiers or directly trained generative models to achieve conditional control, laying the groundwork for conditional diffusion. Conditional diffusion models can also guide the generation process by integrating textual descriptions or other modalities. For example, Zhong et al. [25] combined pre-trained language models with diffusion models to generate high-quality images that align closely with textual descriptions. Similarly, Saharia et al. [26] applied diffusion models to conditional image generation, demonstrating remarkable performance across various image transformation tasks. Furthermore, Liu et al. [27] implemented diffusion processes in latent space for text-to-audio cross-modal generation, significantly reducing computational complexity while preserving generation quality. These advancements have further expanded the application scenarios of diffusion models.

Figure 2: Shows the overall framework of the proposed ADMC method. (a) The independently trained feature extraction networks (ITFN) for each modality, aiming to preserve their unique characteristics. (b) The Attention-based Diffusion Network (ADN), aiming to capture inter-modal dependencies. (c) Multimodal recognition tasks under missing and full-modality scenarios, respectively.

3 METHODOLOGY↩︎

Our method aims to classify the category \(y\) for each video segment, as shown in Fig. 2. We use \(A\), \(V\), and \(T\) to represent the raw acoustic, visual, and textual datas of the video segment, respectively. During the training phase, the model is provided with the full-modalities as input, while in the testing phase, it accepts either partial or full-modalities. The method consists of three stages: First, independently train feature extraction networks for each modality to obtain visual \(F_v\), acoustic \(F_a\), and textual \(F_t\) features (Fig. 2 (a)). Next, fix these feature extraction networks (\(Enc_v\) for visual, \(Enc_a\) for acoustic, and \(Enc_t\) for textual) and train an Attention-based Diffusion Network (ADN) to generate missing modalities features (Fig. 2 (b)). Finally, with the parameters of Enc and ADN fixed, we train Multimodal Fusion classification networks (MF) based on the full-modalities, which outputs the final classification results (Fig. 2 (c)).

3.1 Modality-specific Feature Extraction↩︎

We follow the feature extraction approach described in [7], [11], [28], extracting visual \(F_v\), audio \(F_a\), and text \(F_t\) utterance-level representations from the raw datas. These features are extracted using networks trained on single-modality classification, with early stopping applied to prevent overfitting, as shown in Fig. 2 (a). The parameters of these feature extraction networks are fixed during subsequent tasks. For ease of processing, these features are stacked to form a multimodal feature representation \(F^0_{tva}\), which serves as the input for the following modules.

3.2 Attention-based Diffusion Networks↩︎

The Attention-based Diffusion Networks (ADN) consists of two processes: forward and reverse [27].

In the forward process, multimodal features \(F^0_{tva}\) are gradually corrupted by adding Gaussian noise \(\epsilon\). As the diffusion step \(n \in [1, \dots, N]\) increases, the original information is progressively lost, approaching a noise distribution, as defined in Eq. 1 . Additionally, a multi-head self-attention network[29] is employed to capture inter-modal dependencies and predict the added noise \(\epsilon'\) at each time step, as defined in Eq. 2 . The ADN network is trained by minimizing the mean squared error (MSE) between \(\epsilon\) and \(\epsilon'\), as shown in Fig. 2 (b). \[\label{q951} q(F^n_{tva} | F^0_{tva}) = \mathcal{N}(F^n_{tva}; \sqrt{\bar{\alpha}_n} F^0_{tva}, (1 - \bar{\alpha}_n)\epsilon),\tag{1}\] \[\label{q952} \epsilon' = \text{Attention}(Q^l, K^l, V^l) \quad \text{for } l = 1, 2, \dots, L,\tag{2}\] Where \(F^n_{tva}\) represents the multimodal feature after the \(n\)-th forward diffusion step, and \(\bar{\alpha}_n\) is the cumulative noise scheduling parameter. \(Q^l\), \(K^l\), and \(V^l\) correspond to the query, key, and value vectors of \(F^n_{tva}\), respectively, where \(L\) indicates the number of layers in the self-attention network.

In the reverse process, we incorporate the available modalities part of the latent representation \(F_{o}\) into the generated latent representation \(F_{tva}\). Starting from step \(n\) in the reverse diffusion process, after each inference step (Eq. 3 ), we update \(F^{n-1}_{tva}\) using Eq. 4 . By using this approach, we can generate the missing modality features while preserving the ground-truth data of the available modalities. \[\begin{align} \label{q953} p_\theta(F^{n-1}_{tva} | F^n_{tva}) &= \mathcal{N}\bigg(F^{n-1}_{tva}; \frac{1}{\sqrt{\alpha_n}} \left( F^n_{tva} - \right. \\ &\;\left. \frac{\beta_n}{\sqrt{1 - \bar{\alpha}_n}} \epsilon_\theta(F^n_{tva}, n) \right), \sigma^2_n \mathbf{I} \bigg), \end{align}\tag{3}\] \[\label{q954} F^{n-1}_{tva} = m \odot F^{ n-1}_{o} + (1 - m) \odot F^{n-1}_{tva},\tag{4}\] Where \(F^{n-1}_{tva}\) represents the latent representation after the \((N-n+1)\)-th reverse diffusion step, while \(F^{n-1}_{o}\) denotes the latent representation of the available modality part after the \((n-1)\)-th forward diffusion step. The optimal \(n\) is determined experimentally. The mask \(m\) indicates missing and available modalities, and \(\beta_n\) is the noise scheduling parameter.

3.3 Modality Completion and Enhancement↩︎

We validate our method under both missing modality and full-modality scenarios, as shown in Fig. 2 (c). In the missing modality scenario, we replace the missing features with Gaussian noise \(\epsilon\) and use the trained ADN to generate the missing modality features. This process ensures the completeness of multimodal data, enhancing the model’s robustness across various missing modality scenarios.

In the full-modality scenario, we propose a MultiModal Enhancement Recognition (MMER) method. The input consists of full-modality features, we use the ADN for cross-modal generation (e.g., using \(F_a\) and \(F_v\) to generate \(F'_t\)). These generated features \(F'\) encode key information from other modalities and can provide additional complementary cues to the original modality. To integrate this information, we concatenate \(F\) with \(F'\), resulting in the enhanced modality representation \(F_{\text{enhd}}\). As shown in Eq. 5 , where \(F\) denotes the original modality feature, and \(F'\) represents the generated cross-modal feature obtained from the ADN. The function \(\text{cat}(\cdot)\) denotes the concatenation operation, performed along the channel dimension.

\[\label{q955} F_{\text{enhd}} = \text{cat}(F, F').\tag{5}\]

3.4 Multimodal Feature Fusion Network↩︎

The output feature \(F_M \in \mathbb{R}^{M \times D}\) obtained from completion or enhancement operation is processed by a multi-head self-attention based multimodal fusion network (MF). Specifically, MF first employs the multi-head attention mechanism to model inter-modal dependencies and learn attention weights for each modality. Then, it performs a weighted aggregation of modality features followed by mean pooling along the modality dimension to obtain the fused representation \(u\), as shown in Eq. 6 . Finally, \(u\) is fed into a fully connected classification network to produce the final prediction.

\[u = \text{Mean}\!\left(\text{softmax}\!\left(\frac{QK^\top}{\sqrt{d_k}}\right)V\right), \label{q956}\tag{6}\] where \(Q = F_M W^Q\), \(K = F_M W^K\), and \(V = F_M W^V\). \(W^Q\), \(W^K\) and \(W^V\) are learnable parameters, \(d_k\) denotes the dimension of the key vectors, and Mean(\(\cdot\)) represents the mean pooling operation along the modality dimension.

4 Experiments Setup↩︎

4.1 Dataset↩︎

We evaluated our proposed method on two benchmark multimodal datasets: IEMOCAP [30] and MIntRec [28]. IEMOCAP is an Interactive Emotional dataset with four emotional labels. MIntRec is a multimodal intent recognition dataset collected from life TV dramas, featuring twenty intent labels.

In the IEMOCAP dataset, the frame-level features are 130-dim OpenSMILE features (IS13 ComParE) [31], 342-dim Denseface features from a pretrained DenseNet [32], and 1024-dim BERT-large embeddings [33]. In the MIntRec dataset, the frame-level features are 768-dim wav2vec 2.0 features [34], 256-dim Faster R-CNN features from a pretrained ResNet-50 [28], and 768-dim BERT embeddings [33]. Similar to [7], [11], for \(Enc_a\), we use an LSTM [35] network to capture temporal information from sequential frame-level acoustic features and apply max-pooling to obtain utterance-level acoustic embeddings. For \(Enc_v\), we use the same approach as \(Enc_a\) to obtain utterance-level visual embeddings. For \(Enc_t\), we employ a TextCNN [36] to obtain utterance-level textual embeddings from sequential word-level features.

4.2 Implementation details↩︎

The hidden size for encoders \(Enc_a\) and \(Enc_v\) is set to 256. \(Enc_a\) consists of 3 convolutional blocks with kernel sizes of 3, 4, and 5, each producing an output size of 256. In the ADN module, the self-attention network has 4 encoder layers with 8 attention heads and a hidden size of 1024. The diffusion model uses noise scheduling parameters from \(\beta_s = 0.0003\) to \(\beta_e = 0.06\) over \(N = 1000\) steps. In the MF module, the self-attention network has 2 layers with 4 attention heads, a hidden dimension of 1024, and a dropout rate of 0.5. The classifier consists of 2 fully connected layers of size {128, 4}. We use the Adam optimizer with a dynamic learning rate, starting at 0.0001, and apply the Lambda LR scheduling method for learning rate updates. We select the best model on the validation set and report its performance on the testing set. All models are implemented with Pytorch deep learning toolkit and run on an RTX 4090 GPU.

4.3 Baseline↩︎

The method is compared with recent approaches addressing missing modalities: 1) CRA [5]: is an extension of AE [16]. It combines a series of residual autoencoders into a cascaded architecture for missing data imputation. 2) MCTN [6]: Utilizes cyclic translation between modalities to obtain joint representations. 3) MMIN [7]: Learns joint representations via cross-modality imagination using cascade residual auto-encoder and cycle consistency learning. 4) IF-MMIN [11]: Enhances MMIN by introducing modality-invariant features to reduce modality gaps. 5) IMDer [37]: A score-based diffusion framework that models the distribution of missing modalities and reconstructs them by denoising random noise under semantic conditions. 6) MPLMM [38]: Learning of both intra- and inter-modal information through generative and missing-signal prompts, while significantly reducing the number of trainable parameters. 7) P-RMF [39]: Combines quantified modality intrinsic uncertainty to learn stable multimodal joint representations, enhancing the model’s robustness to noise and missing data.

Table 1: The experimental results of our method with recent other methods on the IEMOCAP (i.e. testing condition {T} indicates that only textual modality is available and both visual and acoustic modalities are missing), Bold numbers indicate the best performance, and methods marked with * have results reproduced using open-source code, while other results are from the original papers.
Methods {A} {V} {T} {AV} {AT} {VT} Average
2-8 WA\(\uparrow\)/UA\(\uparrow\) WA\(\uparrow\)/UA\(\uparrow\) WA\(\uparrow\)/UA\(\uparrow\) WA\(\uparrow\)/UA\(\uparrow\) WA\(\uparrow\)/UA\(\uparrow\) WA\(\uparrow\)/UA\(\uparrow\) WA\(\uparrow\)/UA\(\uparrow\)
CRA (2017) 48.1/49.1 47.7/48.2 62.3/63.5 57.1/54.2 68.7/70.3 68.2/67.7 58.7/58.8
MCTN (2019) 49.7/51.6 48.9/45.7 62.4/63.7 56.3/55.8 68.3/69.4 67.8/68.3 58.9/59.1
MMIN (2021) 56.5/59.0 52.5/51.6 66.5/68.0 63.9/65.4 72.9/75.1 72.6/73.6 64.1/65.2
IF-MMIN (2023) 56.2/58.1 51.9/50.4 67.0/68.2 65.3/66.5 74.0/75.4 72.6/73.6 64.5/65.3
\(\text{IMDer}^{*}\) (2023) 54.1/55.2 55.6/56.3 68.2/68.5 67.1/68.3 71.2/72.3 73.1/73.4 64.8/65.7
MPLMM (2024) 59.7/- 57.6/- 69.2/- 67.2/- 75.9/- 74.6/- 67.4/-
\(\text{P-RMF}^{*}\) (2025) 52.3/56.7 58.6/55.4 68.3/69.6 67.8/68.7 68.9/69.9 73.6/75.4 64.9/66.0
Ours 59.2/61.6 59.8/57.2 71.4/74.1 72.1/72.9 74.1/76.6 75.2/78.5 68.7/70.2
Table 2: The experimental results of our method with recent other methods on the MIntRec.
Methods {A} {V} {T} {AV} {AT} {VT} Average
2-8 WA\(\uparrow\)/UA\(\uparrow\) WA\(\uparrow\)/UA\(\uparrow\) WA\(\uparrow\)/UA\(\uparrow\) WA\(\uparrow\)/UA\(\uparrow\) WA\(\uparrow\)/UA\(\uparrow\) WA\(\uparrow\)/UA\(\uparrow\) WA\(\uparrow\)/UA\(\uparrow\)
CRA (2017) 28.2/17.8 11.9/7.1 68.5/64.6 26.8/19.4 68.7/65.3 67.0/64.1 45.2/39.1
MCTN (2019) 26.5/24.3 13.7/10.1 67.9/62.3 27.7/24.5 68.2/64.1 69.0/64.6 45.5/41.7
MMIN (2021) 24.7/14.6 15.2/6.7 68.3/65.5 25.1/15.6 69.1/65.8 69.4/65.5 45.3/39.0
IF-MMIN (2023) 29.2/23.4 12.8/6.7 67.6/62.1 28.1/23.2 68.3/62.2 67.2/61.8 45.5/39.9
\(\text{IMDer}^{*}\) (2023) 23.6/17.3 10.6/7.9 67.3/62.8 27.6/19.8 69.3/64.2 68.1/63.6 44.4/39.2
\(\text{P-RMF}^{*}\) (2025) 24.8/16.2 11.9/5.3 67.6/64.9 23.7/16.9 70.4/65.3 67.8/65.3 44.6/39.0
Ours 33.4/29.3 15.4/10.2 71.6/65.7 34.3/29.6 72.3/65.8 71.8/65.1 49.8/44.3

5 Experimental results and analysis↩︎

5.1 Comparative Study↩︎

To comprehensively evaluate the effectiveness of our method, we compare it with previous state-of-the-art approaches and report weighted accuracy (WA) and unweighted accuracy (UA), as shown in Tables 3 and  4. On the IEMOCAP dataset, our results significantly outperform previous approaches across all modality-missing scenarios, with an average performance improvement of 1.9%. Similarly, on the MIntRec dataset, our method sustained these advantages, achieving an average performance improvement of 9.4%. These results demonstrate the robustness and generalization of our method in missing modality completion tasks.

Specifically, compared to autoencoder-based methods (AE, CRA), translation-based methods (MCTN, MMIN) typically yield better performance. The modality-to-modality transformations in these methods help capture inter-modality dependencies and generate features for missing modalities. However, IF-MIN and P-RMF consistently underperform ADMC in single-modality input scenarios. This performance gap arises primarily from their joint training strategies, which lead to strong coupling between the feature extraction networks of different modalities. When certain modalities are missing, this coupling causes a significant drop in model performance. Although IMDer also uses diffusion models to reconstruct missing modalities, its performance is limited by the lack of an attention-based cross-modal mechanism. In contrast, ADMC utilizes independently trained feature extraction networks (ITFN), which effectively reduce inter-modality coupling. This approach allows each available modality to fully express its representational power, even when some modalities are missing. Additionally, ADMC employs Attention-based Diffusion Networks (ADN) to capture inter-modality dependencies and generate missing modality features that closely align with the authentic multimodal feature distribution, thereby significantly enhancing the overall performance of the model.

5.2 Ablation Study↩︎

To validate the contributions of the independently trained feature extraction networks (ITFN) and ADN modules, we conducted ablation experiments, as shown in Tables 3 and  4.

Table 3: Ablation study results on IEMOCAP.
Methods {A} {V} {T} {AV} {AT} {VT} Average
2-8 WA\(\uparrow\) WA\(\uparrow\) WA\(\uparrow\) WA\(\uparrow\) WA\(\uparrow\) WA\(\uparrow\) WA\(\uparrow\)
w/o ITFN 57.0 54.7 69.2 67.8 75.6 74.5 66.5
w/o ADN 58.0 58.1 70.0 70.2 73.9 74.3 67.6
W/ U-Net 58.2 57.6 71.4 71.3 72.7 74.5 67.5
Ours 59.2 59.8 71.4 72.1 74.1 75.2 68.7
Table 4: Ablation study results on the MIntRec.
Methods {A} {V} {T} {AV} {AT} {VT} Average
2-8 WA\(\uparrow\) WA\(\uparrow\) WA\(\uparrow\) WA\(\uparrow\) WA\(\uparrow\) WA\(\uparrow\) WA\(\uparrow\)
w/o ITFN 31.6 11.7 71.1 31.8 71.1 71.3 48.1
w/o ADN 32.4 15.0 71.5 33.7 71.5 71.3 49.3
w/ U-Net 32.2 14.5 71.5 33.6 71.1 71.5 49.0
Ours 33.4 15.4 71.6 34.3 72.3 71.8 49.8

1) w/o ITFN: Removing the independently trained feature extraction networks and using the jointly trained extractor, while missing modality features are generated by the ADN, leads to a performance drop under missing-modality conditions (e.g., using only audio or only visual modality). This indicates that joint training leads to over-coupled feature extraction across modalities, making the model less robust when certain modalities are missing. 2) w/o ADN: Retaining the independent feature extractors but replacing missing modality features with zero vectors instead of reconstructing them via ADN causes performance degradation across all scenarios, showing that without reconstructing missing modalities, multimodal information cannot be fully utilized. With ADN, the model can generate complementary features based on the available modalities, thereby improving robustness under missing-modality conditions. 3) w/ U-Net: Replacing the self-attention module in ADN with a standard U-Net[40] yields lower performance. Although U-Net can generate features close to the target distribution, it lacks explicit modeling of cross-modal dependencies, resulting in reconstructed features that are less consistent with the global multimodal semantics and thus lower overall performance. In contrast, ADN captures cross-modal relationships through self-attention and produces more discriminative complementary features.

5.3 Visualization Analysis↩︎

To intuitively analyze the impact of different values of \(n\) on the quality of generated features during reverse generation, we visualized the differences between ground-truth multimodal embeddings and the corresponding generated embeddings. Specifically, we randomly selected 600 real samples and their generated counterparts from each modality, and applied the T-SNE toolkit [41] for dimensionality reduction. As shown in Figure 3, we observe that: 1) Features from different modalities exhibit clear clustering patterns in the embedding space, indicating that the ground-truth embeddings effectively preserve modality-specific information. 2) Extremely small or large values of \(n\) negatively impact the quality of the generated embeddings, leading to blurred distribution boundaries and overlaps between modalities. 3) When \(n\) = 50, the generated embeddings closely align with the distribution of the ground-truth embeddings, demonstrating the effectiveness of our method in generating missing modality features. 4) On the MIntRec dataset, the generated features show a higher degree of dispersion, primarily due to the limited sample size, which restricts the model’s ability to capture the underlying data distribution.

a
b

Figure 3: Shows the visualization of ground-truth multimodal embeddings and generated embeddings on the IEMOCAP and MIntRec dataset. Blue represents text feature embeddings, green represents audio feature embeddings, and yellow represents visual feature embeddings.. a — Visualization on the IEMOCAP dataset., b — Visualization on the MIntRec dataset.

5.4 Effects of MMER↩︎

To evaluate the effectiveness of the MMER module, we conducted comparative experiments under full-modality scenarios, as shown in Table 5. The experiment compared three methods: unrestricted feature extraction networks (Jointly_training), fixed feature extraction networks (Use_ITFN), and using generated features to enhance original modality representations (MMER). The results indicate that the Jointly_Training method outperforms the Use_ITFN method by dynamically adjusting the feature representations of each modality, enabling the model to achieve optimal performance. The MMER method achieves the best performance by enhancing original modality representations using generated features.

Existing methods for addressing missing modality problems often perform poorly in full-modality scenarios. In contrast, our method not only focuses on tackling missing modality issues but also demonstrates excellent performance in full-modality tasks.

Table 5: Result of the study on the MMER.
Methods IEMOCAP MIntRec
2-5 WA\(\uparrow\) UA\(\uparrow\) WA\(\uparrow\) UA\(\uparrow\)
Jointly_training 77.4 79.1 71.2 65.0
Use_ITFN 76.5 78.1 71.5 65.0
MMER 77.3 79.5 72.5 65.4

5.5 Experimental of Different Missing Rate↩︎

To evaluate the robustness of multimodal models under inter-modality missingness, we using a probabilistic approach to randomly drop one or two modalities at varying missing ratios \(p \in {0.1, 0.2, \ldots, 0.5}\). The performance trends of different models under increasing missing ratios are shown in Fig. 4.

Figure 4: Results under different missing rates on the IEMOCAP dataset.

As the missing ratio increases, the F1 scores of all baseline methods consistently decline, indicating that inter-modality missingness disrupts semantic integrity and weakens the effectiveness of multimodal feature fusion. In contrast, our method achieves the best performance across all missing ratios and suffers the smallest performance degradation. This advantage stems, on the one hand, from the ADN module, which generates missing modality features that closely match the real feature distribution and thus enhance multimodal fusion, and on the other hand from the decoupled feature extraction mechanism in our framework, which effectively mitigates the impact of missing modalities on overall model performance.

5.6 Time Complexity and Inference Cost↩︎

Introducing a diffusion model inevitably increases inference time. We therefore analyze the computational complexity and practical inference cost of the proposed ADN module. ADN is a 4-layer Transformer-encoder-based self-attention network. For an input with \(L\) modalities and feature dimension \(D\), the per-step inference complexity is \(\mathcal{O}(L^2 D + L D^2)\). In our setting, the number of modalities is very small (\(L = 3\)), so the overall complexity is substantially lower than that of standard long-sequence Transformers.

We further evaluate the actual inference latency on an RTX 4090 GPU. With 1,000 diffusion denoising steps, the additional inference time introduced by ADN is approximately 50 ms per sample, which is acceptable for our target scenarios. Moreover, existing acceleration techniques for diffusion models (e.g., DDIM, reduced-step samplers, and step-skipping strategies) can reduce the number of steps to 50–100 with negligible performance degradation, indicating clear potential for further efficiency improvements.

6 CONCLUSION↩︎

In this paper, we propose an Attention-based Diffusion Model for Missing modalities feature Completion (ADMC), which integrates independently trained feature extraction networks and an Attention-based Diffusion Network (ADN). Our approach preserves the distinct characteristics of each modality while generating missing modality features that closely adhere to the authentic multimodal feature distribution. Extensive experimental results demonstrate that our method achieves state-of-the-art performance. In future work, we will continue to explore the applications of attention-based diffusion networks in a broader range of tasks.

7 Safe and Responsible Innovation Statement↩︎

This work supports more robust multimodal emotion and intent recognition under missing-modality conditions, which may improve accessibility and reliability in human-computer interaction. However, such systems may also introduce risks related to privacy, bias, and misuse. The datasets used in this study are public benchmarks, and we do not collect new personal data. We acknowledge that emotion and intent recognition models can inherit demographic, cultural, and contextual biases from training data, potentially leading to unfair outcomes. Our method is intended as a decision-support tool rather than a sole basis for high-stakes decisions. Responsible deployment should include human oversight, bias evaluation across groups, privacy protection, and clear usage boundaries.

References↩︎

[1]
Z. Li, F. Tang, M. Zhao, and Y. Zhu, “EmoCaps: Emotion capsule based model for conversational emotion recognition,” arXiv preprint arXiv:2203.13504, 2022.
[2]
X. Huang, T. Ma, L. Jia, Y. Zhang, H. Rong, and N. Alnabhan, “An effective multimodal representation and fusion method for multimodal intent recognition,” Neurocomputing, vol. 548, p. 126373, 2023.
[3]
Q. Zhou et al., “Token-level contrastive learning with modality-aware prompting for multimodal intent recognition,” in Proceedings of the AAAI conference on artificial intelligence, 2024, vol. 38, pp. 17114–17122.
[4]
L. Cai, Z. Wang, H. Gao, D. Shen, and S. Ji, “Deep adversarial learning for multi-modality missing data completion,” in Proceedings of the 24th ACM SIGKDD international conference on knowledge discovery & data mining, 2018, pp. 1158–1166.
[5]
L. Tran, X. Liu, J. Zhou, and R. Jin, “Missing modalities imputation via cascaded residual autoencoder,” in Computer vision & pattern recognition, 2017, pp. 4971–4980.
[6]
H. Pham, P. P. Liang, T. Manzini, L.-P. Morency, and B. Póczos, “Found in translation: Learning robust joint representations by cyclic translations between modalities,” in Proceedings of the AAAI conference on artificial intelligence, 2019, vol. 33, pp. 6892–6899.
[7]
J. Zhao, R. Li, and Q. Jin, “Missing modality imagination network for emotion recognition with uncertain missing modalities,” in Proceedings of the 59th annual meeting of the association for computational linguistics and the 11th international joint conference on natural language processing (volume 1: Long papers), 2021, pp. 2608–2618.
[8]
J. Zeng, T. Liu, and J. Zhou, “Tag-assisted multimodal sentiment analysis under uncertain missing modalities,” in Proceedings of the 45th international ACM SIGIR conference on research and development in information retrieval, 2022, pp. 1545–1554.
[9]
J. Zeng, J. Zhou, and T. Liu, “Mitigating inconsistencies in multimodal sentiment analysis under uncertain missing modalities,” in Proceedings of the 2022 conference on empirical methods in natural language processing, 2022, pp. 2924–2934.
[10]
Z. Wang, Z. Wan, and X. Wan, “Transmodality: An end2end fusion method with transformer for multimodal sentiment analysis,” in Proceedings of the web conference 2020, 2020, pp. 2514–2520.
[11]
H. Zuo, R. Liu, J. Zhao, G. Gao, and H. Li, “Exploiting modality-invariant feature for robust multimodal emotion recognition with missing modalities,” in ICASSP 2023-2023 IEEE international conference on acoustics, speech and signal processing (ICASSP), 2023, pp. 1–5.
[12]
S. Parthasarathy and S. Sundaram, “Training strategies to handle missing modalities for audio-visual expression recognition,” in Companion publication of the 2020 international conference on multimodal interaction, 2020, pp. 400–404.
[13]
C. Zhang, Y. Cui, Z. Han, J. T. Zhou, H. Fu, and Q. Hu, “Deep partial multi-view learning,” IEEE transactions on pattern analysis and machine intelligence, vol. 44, no. 5, pp. 2402–2415, 2020.
[14]
H. Fan, Y. Chen, Y. Guo, H. Zhang, and G. Kuang, “Hyperspectral image restoration using low-rank tensor recovery,” IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, vol. 10, no. 10, pp. 4589–4604, 2017.
[15]
P. Liang, Z. Liu, Y. Tsai, Q. Zhao, R. Salakhutdinov, and L. Morency, “Learning representations from imperfect time series data via tensor rank regularization. arXiv 2019,” arXiv preprint arXiv:1907.01011, 2019.
[16]
Y. Duan, Y. Lv, W. Kang, and Y. Zhao, “A deep learning based approach for traffic data imputation,” in Proceedings of the 17th international IEEE conference on intelligent transportation systems (ITSC), 2014, pp. 912–917.
[17]
L. Cai, Z. Wang, H. Gao, D. Shen, and S. Ji, “Deep adversarial learning for multi-modality missing data completion,” in Proceedings of the 24th ACM SIGKDD international conference on knowledge discovery & data mining, 2018, pp. 1158–1166.
[18]
C. Du et al., “Semi-supervised deep generative modelling of incomplete multi-modality emotional data,” in Proceedings of the 26th ACM international conference on multimedia, 2018, pp. 108–116.
[19]
Z. Yuan, W. Li, H. Xu, and W. Yu, “Transformer-based feature reconstruction network for robust multimodal sentiment analysis,” in Proceedings of the 29th ACM international conference on multimedia, 2021, pp. 4400–4407.
[20]
Z. Lian, L. Chen, L. Sun, B. Liu, and J. Tao, “GCNet: Graph completion network for incomplete multimodal learning in conversation,” IEEE Transactions on pattern analysis and machine intelligence, vol. 45, no. 7, pp. 8419–8432, 2023.
[21]
J. Ho, A. Jain, and P. Abbeel, “Denoising diffusion probabilistic models,” Advances in neural information processing systems, vol. 33, pp. 6840–6851, 2020.
[22]
Y. Song, J. Sohl-Dickstein, D. P. Kingma, A. Kumar, S. Ermon, and B. Poole, “Score-based generative modeling through stochastic differential equations.” 2021, [Online]. Available: https://arxiv.org/abs/2011.13456.
[23]
A. Q. Nichol and P. Dhariwal, “Improved denoising diffusion probabilistic models,” in International conference on machine learning, 2021, pp. 8162–8171.
[24]
J. Ho and T. Salimans, “Classifier-free diffusion guidance,” arXiv preprint arXiv:2207.12598, 2022.
[25]
S. Zhong, Z. Huang, W. Wen, J. Qin, and L. Lin, “Sur-adapter: Enhancing text-to-image pre-trained diffusion models with large language models,” in Proceedings of the 31st ACM international conference on multimedia, 2023, pp. 567–578.
[26]
C. Saharia et al., “Palette: Image-to-image diffusion models,” in ACM SIGGRAPH 2022 conference proceedings, 2022, pp. 1–10.
[27]
H. Liu et al., “AudioLDM: Text-to-audio generation with latent diffusion models,” arXiv preprint arXiv:2301.12503, 2023.
[28]
H. Zhang, H. Xu, X. Wang, Q. Zhou, S. Zhao, and J. Teng, “Mintrec: A new dataset for multimodal intent recognition,” in Proceedings of the 30th ACM international conference on multimedia, 2022, pp. 1688–1697.
[29]
A. Vaswani et al., “Attention is all you need,” in Advances in neural information processing systems, 2017, vol. 30.
[30]
C. Busso et al., “IEMOCAP: Interactive emotional dyadic motion capture database,” Language resources and evaluation, vol. 42, pp. 335–359, 2008.
[31]
F. Eyben, M. Wöllmer, and B. Schuller, “Opensmile: The munich versatile and fast open-source audio feature extractor,” in Proceedings of the 18th ACM international conference on multimedia, 2010, pp. 1459–1462.
[32]
G. Huang, Z. Liu, L. Van Der Maaten, and K. Q. Weinberger, “Densely connected convolutional networks,” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2017, pp. 4700–4708.
[33]
J. Devlin, M.-W. Chang, K. Lee, and K. Toutanova, “Bert: Pre-training of deep bidirectional transformers for language understanding. arXiv,” arXiv preprint arXiv:1810.04805, 2019.
[34]
A. Baevski, Y. Zhou, A. Mohamed, and M. Auli, “wav2vec 2.0: A framework for self-supervised learning of speech representations,” Advances in neural information processing systems, vol. 33, pp. 12449–12460, 2020.
[35]
H. Sak, A. Senior, and F. Beaufays, “Long short-term memory based recurrent neural network architectures for large vocabulary speech recognition,” arXiv preprint arXiv:1402.1128, 2014.
[36]
Y. Kim, “Convolutional neural networks for sentence classification,” in Proceedings of the 2014 conference on empirical methods in natural language processing (EMNLP), 2014, pp. 1746–1751.
[37]
Y. Wang, Y. Li, and Z. Cui, “Incomplete multimodality-diffused emotion recognition,” Advances in Neural Information Processing Systems, vol. 36, pp. 17117–17128, 2023.
[38]
Z. Guo, T. Jin, and Z. Zhao, “Multimodal prompt learning with missing modalities for sentiment analysis and emotion recognition,” arXiv preprint arXiv:2407.05374, 2024.
[39]
A. Zhu, M. Hu, X. Wang, J. Yang, Y. Tang, and N. An, “Proxy-driven robust multimodal sentiment analysis with incomplete data,” in Proceedings of the 63rd annual meeting of the association for computational linguistics (volume 1: Long papers), 2025, pp. 22123–22138.
[40]
O. Ronneberger, P. Fischer, and T. Brox, “U-net: Convolutional networks for biomedical image segmentation,” in International conference on medical image computing and computer-assisted intervention, 2015, pp. 234–241.
[41]
L. Van der Maaten and G. Hinton, “Visualizing data using t-SNE,” Journal of Machine Learning Research, vol. 9, pp. 2579–2605, 2008.