Training diffusion models for audiovisual sequences allows for a range of generation tasks by learning conditional distributions of various input-output combinations of the two modalities. Nevertheless, this strategy often requires training a separate model for each task which is expensive. Here, we propose a novel training approach to effectively learn arbitrary conditional distributions in the audiovisual space.Our key contribution lies in how we parameterize the diffusion timestep in the forward diffusion process. Instead of the standard fixed diffusion timestep, we propose applying variable diffusion timesteps across the temporal dimension and across modalities of the inputs. This formulation offers flexibility to introduce variable noise levels for various portions of the input, hence the term mixture of noise levels. We propose a transformer-based audiovisual latent diffusion model and show that it can be trained in a task-agnostic fashion using our approach to enable a variety of audiovisual generation tasks at inference time. Experiments demonstrate the versatility of our method in tackling cross-modal and multimodal interpolation tasks in the audiovisual space. Notably, our proposed approach surpasses baselines in generating temporally and perceptually consistent samples conditioned on the input. Project page: avdit2024.github.io
Paper
References (54)
Scroll for more · 38 remaining
Similar papers
Peer review
Summary
This paper presents a new diffusion model that is capable of generating data following various conditional distributions in multimodal data. Rather than applying a uniform noise level to all data elements, the proposed model permits the use of differing noise levels across modalities and time dimensions. This allows the model to concurrently learn a variety of conditional distributions, including cross-modal conditioning, temporal interpolation, and temporal continuation. The experimental results with audio-video datasets validate the effectiveness of the proposed model.
Strengths
- The idea of MoNL is simple and reasonable. It definitely enables the model to effectively learn various conditional distributions simultaneously. - The proposed method performs well in the experiments. MoNL particularly contributes to boost the performance of the model in AV-inpaint and AV-continue tasks. - The manuscript is well-written and easy to follow.
Weaknesses
- The major concern is on the experiments: the majority of the empirical evaluations are conducted with the internal dataset. As there are already several publicly-available datasets (such as Landscape, AIST++, and VGGSound) that are commonly used in the literature, I strongly encourage the authors to do at least the ablation studies with such datasets to make them reproducible. - The novelty in methoology is somewhat limited. The idea of using various noise levels in multimodal generative modeling has been presented in the prior work [3]. A particular challenge in this paper is to extend this idea also for modeling temporal dynamics, but similar strategies are commonly adopted in several video generation studies: - “Diffusion Models for Video Prediction and Infilling,” TMLR 2022. - “MCVD: Masked Conditional Video Diffusion for Prediction, Generation, and Interpolation,” NeurIPS 2022.
Questions
- In the ablations at Table 1, why is Vanilla not used? Without the Vanilla strategy, the model may not encounter cases where all elements are almost entirely noise or close to clean ones during training, which could potentially degrade the performance of the model to some extent. Therefore, a more reasonable approach for ablation might be to gradually add other strategies to Vanilla. --- <post-rebuttal> I have updated my rating from 4 to 5, as my concern on the dataset is somewhat resolved through the discussion.
Rating
5
Confidence
4
Soundness
2
Presentation
4
Contribution
2
Limitations
Limitations are properly discussed in the appendix.
Theoretical Background on Mixture of Noise Levels
## Theoretical Background on Mixture of Noise Levels $ $ ### 1. Theoretical Background on Multimodal Learning In *"A Theory of Multimodal Learning"* [1], multimodal learning is shown to offer a superior **generalization bound** compared to unimodal learning, with an improvement factor of **$O(\sqrt{n})$**, where **$n$** denotes the sample size. This benefit relies on **connection** and **heterogeneity** between modalities: - **Connection**: The bound depends on learned connections between (**$\mathcal{X}$**) and (**$\mathcal{Y}$**). - **Heterogeneity**: Describes how modalities, **$\mathcal{X}$** and **$\mathcal{Y}$**, diverge and complement. If connection and heterogeneity are missing, ill-conditioned scenarios can arise. For instance, if **$x \equiv y$**, perfect connection suggests no need for learning about **$\mathcal{Y}$**. On the other hand, if **$x$** is random noise, there is heterogeneity but no meaningful connection between **$\mathcal{X}$** and **$\mathcal{Y}$**, making non-trivial learning on **$\mathcal{X}$** alone impractical. The theory also highlights that and learning effective connections between modalities via **generative models** can enhance multimodal learning. This forms the basis for our **Mixture of Noise Levels (MoNL)** approach, which is particularly suited for multimodal learning with **audio** and **video** data. [1] Zhou Lu. *"A Theory of Multimodal Learning."* NeurIPS 2023. $ $ ### 2. Advantages of Mixture of Noise Level Training (MoNL) Our **Mixture of Noise Level (MoNL)** training method offers significant benefits for multimodal learning, especially with **audio** and **video** data: - **Heterogeneity and Connection**: Audio and video are naturally heterogeneous. For example, a video of a person speaking includes **audio** of spoken words and **video** of lip movements and facial expressions. MoNL uses **variable noise levels** to enhance learning by capturing the **generic transition matrix** across the **temporal axis**. $$p _{\mathbf{\theta}}([\mathbf{z} _{t ^{(1,1)}-1} ^{(1,1)}, \ldots, \mathbf{z} _{t ^{(M, N)}-1} ^{(M, N)}] \mid[\mathbf{z} _{t ^{(1,1)}} ^{(1,1)}, \ldots, \mathbf{z} _{t ^{(M, N)}} ^{(M, N)}]) \qquad \text{(Eq. (4))}$$ where $M$, $N$ are the number of modalities and time-segments, respectively. - **Enhanced Connectivity**: MoNL improves **connectivity** between **audio** and **video** modalities. Our experiments show that MoNL often surpasses **task-specific learning** approaches by fostering better connections between modalities, adapting its focus more effectively. $ $ ### 3. Enhanced Connectivity - Comparison with Existing Methods - **MoNL vs. Joint Learning in MMD** [2]: Unlike joint learning methods that focus on the joint distribution $p_{\mathbf{\theta}}(\mathbf{z} _{t-1} \mid\mathbf{z} _{t})$, MoNL trains across **multiple conditioning**, enabling better connections by varying its focus. This is evidenced by MoNL outperforming the Vanilla (see Table 1) and MMD models (see Tables 2 and 3). - **MoNL vs. Per-Modality Training**: MoNL goes beyond per-modality training in UniDiffuser [3], which uses variable noise between modalities i.e., learning $p_{\mathbf{\theta}}([\mathbf{z} _{t ^{(1)}-1} ^{(1)}, \ldots, \mathbf{z} _{t ^{(M)}-1} ^{(M)}] \mid[\mathbf{z} _{t ^{(1)}} ^{(1)}, \ldots, \mathbf{z} _{t ^{(M)}} ^{(M)}])$. MoNL introduces variable noise across different **time segments**, learning connections across **temporal dynamics** as well. This advantage is demonstrated in Table 1. - **MoNL vs. Masked Training** [4]: Diffusion models often obscure high-frequency details with low noise and low-frequency structures with high noise [5]. MoNL employs variable noise levels to explore diverse **frequency components**, enhancing the model's ability to correlate high and low-frequency elements. This is in contrast to masked self-supervised learning, which limits frequency-specific connections by masking entire elements. [2] Ruan et al. *"MM-Diffusion: Learning Multi-Modal Diffusion Models for Joint Audio and Video Generation."* CVPR 2023. [3] Bao et al. *"One Transformer Fits All Distributions in Multi-Modal Diffusion at Scale."* ICML 2023. [4] Voleti et al. *"MCVD: Masked Conditional Video Diffusion for Prediction, Generation, and Interpolation."* NeurIPS 2022. [5] Sander Dieleman. *"Noise Schedules Considered Harmful."* [Link](https://sander.ai/2024/06/14/noise-schedules). $ $ ### Conclusion In summary, the effectiveness of **MoNL** for **multimodal diffusion models**, particularly with **audio** and **video** data, stems from its strategic use of **connection** and **heterogeneity**. By applying **variable noise levels**, MoNL enhances **connectivity** between modalities and better adapts to diverse **temporal** and **frequency components**, leading to superior performance compared to existing multimodal learning methods.
Clarification on Resource Constraints and Additional Theoretical Contributions
Dear Reviewer 4xRp, Thank you for your feedback and for considering our initial response. We acknowledge the importance of validating our model on publicly available datasets. Unfortunately, training on these datasets with the necessary computational resources (requiring approximately $35k per model) presented a significant challenge at the moment. However, we plan to proceed the experiments as the permit is approved and believe our work with the Monologues dataset, which contains 16M samples with a diverse range of audiovisual cues, provides valuable insights and demonstrates the robustness of our approach. In addition, we have newly provided the **"Theoretical Background on Mixture of Noise Levels,"** which we hope you will consider as part of our ongoing effort to strengthen our work and contribute to the field. We hope you understand our constraints and continue to see the value in the contributions we have made with the resources available. We remain committed to advancing research in **audiovisual generation** and plan to release the Monologues dataset to further support reproducibility in the future and encourage future work. Thank you for your consideration!
Thanks for further clarification
I appreciate that the authors plan to release the Monologues dataset. I would like to hear more on how the dataset was properly constructed to avoid any issues related to license or private information. One quick comment on the additional theoretical background. This theoretical perspective itself sounds really interesting. While I have no doubt on heterogeneity of audio and video, it seems that it is not trivial to show how MoNL contributes to boost connectivity in theory.
Clarifications and Additional Insights on the Monologues Dataset and MoNL
Dear Reviewer 4xRp, Thank you for your valuable feedback. We appreciate your interest in the Monologues dataset and your insights into the theoretical aspects of our work. **Dataset Construction** To address your concerns about dataset construction, we want to emphasize that the Monologues dataset was meticulously curated to adhere to ethical and legal standards. We have implemented robust measures to protect user privacy and avoid copyright infringement. The dataset consists exclusively of public links (and timestamps) to videos that are: * Including those that are **older than 90 days** * Free from **copyright claims** * Excluding **harmful or inappropriate material** (e.g., nudity, violence) * Verified to contain **motion** (excluding static images) * In the **English language** with a **single visible speaker** * Framed with **torso up** The dataset link list is automatically updated on a daily basis to ensure ongoing compliance with these criteria. The Monologues dataset is a valuable resource for the research community, providing a rich audiovisual benchmark for developing and evaluating multimodal models. **Theoretical Background** While formally proving the connectivity benefits of MoNL presents a significant challenge, our empirical results provide compelling evidence of its effectiveness. Table 1 and Figures 2 and 7 demonstrate MoNL's superior performance across diverse audiovisual tasks, generating more temproally-consistent audiovisuals compared to joint learning and per-modality method, suggesting enhanced intra- and inter-modality connectivity. We recognize the need for a deeper theoretical understanding and plan to explore this area in future research. As highlighted in *"A Theory of Multimodal Learning."* (Zhou et al. NeurIPS 2023), theoretical foundations in multimodal learning remain relatively underdeveloped. Our work serves as a **crucial experimental foundation** for future theoretical investigations in this field. We believe our work represents a **substantial step forward in audiovisual and multimodal research**. **We believe these clarifications provide valuable insights into our work and would be grateful for your consideration of an improved score.** Sincerely, The Authors
Additional analysis of the objectives and effectiveness of our approach
To enhance the understanding of how MoNL contributes to building a unified model for diverse audiovisual tasks—including cross-modal inference and multimodal interpolation— and improve connectivity between modalities, we provide a detailed analysis of the objectives and effectiveness of our approach, considering a simplified case with two modalities and two time-segments. $ $ ### Objective Functions To determine the optimal objective function for learning a model $\theta$ that estimates: - **Cross-modal inferences:** $P(X|Y)$ and $P(Y|X)$ - **Multimodal interpolations:** $P([\mathbf{x}_2, \mathbf{y}_2] \mid [\mathbf{x}_1, \mathbf{y}_1])$, $P([\mathbf{x}_1, \mathbf{y}_1] \mid [\mathbf{x}_2, \mathbf{y}_2])$, $P([\mathbf{x}_1, \mathbf{y}_2] \mid [\mathbf{x}_2, \mathbf{y}_1])$, and $P([\mathbf{x}_2, \mathbf{y}_1] \mid [\mathbf{x}_1, \mathbf{y}_2])$ we evaluate the following objectives for heterogeneous multimodal data $X$ and $Y$, where $X = [\mathbf{x}_1, \mathbf{x}_2]$ and $Y = [\mathbf{y}_1, \mathbf{y}_2]$. 1. **Task-Specific Method** $$ \min \mathbb{E}[P_\theta(X \mid Y)] $$ This approach focuses solely on one direction of inference, potentially missing significant interactions between modalities. 2. **Joint Learning Method** $$ \min \mathbb{E}[P_\theta(X, Y)] $$ This method captures both modalities simultaneously but may fail to model temporal and multimodal interactions, which are crucial for conditional tasks. 3. **Per-Modality Method** $$ \min \mathbb{E}\left[P_\theta(X, Y) + P_\theta(X \mid Y) + \alpha P_\theta(Y \mid X)\right] $$ This method improves upon the task-specific approach by incorporating reverse inference between modalities. However, it may not effectively capture temporal interactions, which are important for interpolation tasks. 4. **Mixture of Noise Levels (MoNL)** $$ \min \mathbb{E}\left[P_\theta(X, Y) + \beta_1 P_\theta(X \mid Y) + \beta_2 P_\theta(Y \mid X) + \beta_3 P([\mathbf{x}_2, \mathbf{y}_2] \mid [\mathbf{x}_1, \mathbf{y}_1]) + \beta_4 P([\mathbf{x}_1, \mathbf{y}_1] \mid [\mathbf{x}_2, \mathbf{y}_2]) + \beta_5 P([\mathbf{x}_1, \mathbf{y}_2] \mid [\mathbf{x}_2, \mathbf{y}_1]) + \beta_6 P([\mathbf{x}_2, \mathbf{y}_1] \mid [\mathbf{x}_1, \mathbf{y}_2])\right] $$ MoNL integrates joint distributions, cross-modal inferences, and multimodal interpolations by learning a transition matrix that captures complex interactions between modalities. $ $ ### Heterogeneous and Connected Multimodality For heterogeneous multimodal data (e.g., audio and video), where $X$ and $Y$ are interconnected but distinct, MoNL excels by capturing intricate interactions through: - **Direct Relationships:** Modeled by $P_\theta(X, Y)$ - **Cross-Modal Inferences:** Modeled by $P_\theta(X \mid Y)$ and $P_\theta(Y \mid X)$ - **Multimodal Interpolations:** Modeled by $P([\mathbf{x}_2, \mathbf{y}_2] \mid [\mathbf{x}_1, \mathbf{y}_1])$, $P([\mathbf{x}_1, \mathbf{y}_1] \mid [\mathbf{x}_2, \mathbf{y}_2])$ and similar terms. MoNL effectively models the connectivity and complex interactions between modalities, significantly enhancing performance in various audiovisual tasks. $ $ We hope this analysis is helpful and will be incorporated into the final version.
Thanks for the response
Thanks for the response. I will update my rating after the discussion with the other reviwers.
Summary
The paper tackles the audio-visual cross-modality generation problem and proposes a training approach to learn arbitrary conditional distributions in the audiovisual space. At the methodological level, the authors propose to apply variable diffusion timesteps across the temporal dimension. The experiments are conducted on Monologues, AIST++, and Landscape datasets.
Strengths
The high-level motivation to learn arbitrary conditional distributions in the audiovisual space is interesting. The experiments show promising results and the authors conduct user studies as supplementary evaluations.
Weaknesses
1. The writing and presentation of the paper can be improved. Figure 1 seems to have some issues with the first-row caption for the AIST dataset, which makes it difficult to read. The usage of math symbols is inconsistent, e.g., the $x$ should be $\bf{x}$ in Line 72. The term “multivariate data” in Line 63 is also not rigorous and confusing, in other words, do the authors imply that the static image data is uni-variante? The use of $\equiv$ and $=$ is also mixed in the paper. 2. One of the key claims and motivations, “Training separate models for each variation is expensive and impractical”, seems controversial and susceptible to me, while I understand training separate DMs for audio and visual data could be expensive, I don’t think that learning a single mixture is a better option than learning two separate data distributions, because learning a mixture increase the complexity of target data distribution, and thus should not be beneficial for performance especially if the domain gap between separate modality is large. As this is one of the key claims that motivates the methodological design, I was expecting more rigorous theoretical support in the paper, could the authors elaborate on that? 3. For the experiments, while the authors claim that “the sequential representations can be either latent spaces or raw data”, the actual implementation only conducts experiments on low dimensional features of pre-trained models, i.e., MAGVIT-v2 for video representations and SoundStream for audio representations. While this is not a serious flaw, it raises questions about the generalizability of the proposed method, and the fairness in comparison with other methods, as the pre-trained models may (and very likely) directly influence the final performance. And the baseline method MM-Diffusion seems not to operate on the same feature space according to Appendix C?
Questions
In addition to the comments in the Weaknesses section. I may have missed several details: how is the scheduler defined in the proposed method Eq. (5)? What is the time cost for training the proposed model? And how does this design yield a difference compared to training on single modality or separate DMs?
Rating
5
Confidence
4
Soundness
3
Presentation
2
Contribution
2
Limitations
The limitations are discussed in the Appendix A.
Theoretical Background on Mixture of Noise Levels
## Theoretical Background on Mixture of Noise Levels $ $ ### 1. Theoretical Background on Multimodal Learning In *"A Theory of Multimodal Learning"* [1], multimodal learning is shown to offer a superior **generalization bound** compared to unimodal learning, with an improvement factor of **$O(\sqrt{n})$**, where **$n$** denotes the sample size. This benefit relies on **connection** and **heterogeneity** between modalities: - **Connection**: The bound depends on learned connections between (**$\mathcal{X}$**) and (**$\mathcal{Y}$**). - **Heterogeneity**: Describes how modalities, **$\mathcal{X}$** and **$\mathcal{Y}$**, diverge and complement. If connection and heterogeneity are missing, ill-conditioned scenarios can arise. For instance, if **$x \equiv y$**, perfect connection suggests no need for learning about **$\mathcal{Y}$**. On the other hand, if **$x$** is random noise, there is heterogeneity but no meaningful connection between **$\mathcal{X}$** and **$\mathcal{Y}$**, making non-trivial learning on **$\mathcal{X}$** alone impractical. The theory also highlights that and learning effective connections between modalities via **generative models** can enhance multimodal learning. This forms the basis for our **Mixture of Noise Levels (MoNL)** approach, which is particularly suited for multimodal learning with **audio** and **video** data. [1] Zhou Lu. *"A Theory of Multimodal Learning."* NeurIPS 2023. $ $ ### 2. Advantages of Mixture of Noise Level Training (MoNL) Our **Mixture of Noise Level (MoNL)** training method offers significant benefits for multimodal learning, especially with **audio** and **video** data: - **Heterogeneity and Connection**: Audio and video are naturally heterogeneous. For example, a video of a person speaking includes **audio** of spoken words and **video** of lip movements and facial expressions. MoNL uses **variable noise levels** to enhance learning by capturing the **generic transition matrix** across the **temporal axis**. $$p _{\mathbf{\theta}}([\mathbf{z} _{t ^{(1,1)}-1} ^{(1,1)}, \ldots, \mathbf{z} _{t ^{(M, N)}-1} ^{(M, N)}] \mid[\mathbf{z} _{t ^{(1,1)}} ^{(1,1)}, \ldots, \mathbf{z} _{t ^{(M, N)}} ^{(M, N)}]) \qquad \text{(Eq. (4))}$$ where $M$, $N$ are the number of modalities and time-segments, respectively. - **Enhanced Connectivity**: MoNL improves **connectivity** between **audio** and **video** modalities. Our experiments show that MoNL often surpasses **task-specific learning** approaches by fostering better connections between modalities, adapting its focus more effectively. $ $ ### 3. Enhanced Connectivity - Comparison with Existing Methods - **MoNL vs. Joint Learning in MMD** [2]: Unlike joint learning methods that focus on the joint distribution $p_{\mathbf{\theta}}(\mathbf{z} _{t-1} \mid\mathbf{z} _{t})$, MoNL trains across **multiple conditioning**, enabling better connections by varying its focus. This is evidenced by MoNL outperforming the Vanilla (see Table 1) and MMD models (see Tables 2 and 3). - **MoNL vs. Per-Modality Training**: MoNL goes beyond per-modality training in UniDiffuser [3], which uses variable noise between modalities i.e., learning $p_{\mathbf{\theta}}([\mathbf{z} _{t ^{(1)}-1} ^{(1)}, \ldots, \mathbf{z} _{t ^{(M)}-1} ^{(M)}] \mid[\mathbf{z} _{t ^{(1)}} ^{(1)}, \ldots, \mathbf{z} _{t ^{(M)}} ^{(M)}])$. MoNL introduces variable noise across different **time segments**, learning connections across **temporal dynamics** as well. This advantage is demonstrated in Table 1. - **MoNL vs. Masked Training** [4]: Diffusion models often obscure high-frequency details with low noise and low-frequency structures with high noise [5]. MoNL employs variable noise levels to explore diverse **frequency components**, enhancing the model's ability to correlate high and low-frequency elements. This is in contrast to masked self-supervised learning, which limits frequency-specific connections by masking entire elements. [2] Ruan et al. *"MM-Diffusion: Learning Multi-Modal Diffusion Models for Joint Audio and Video Generation."* CVPR 2023. [3] Bao et al. *"One Transformer Fits All Distributions in Multi-Modal Diffusion at Scale."* ICML 2023. [4] Voleti et al. *"MCVD: Masked Conditional Video Diffusion for Prediction, Generation, and Interpolation."* NeurIPS 2022. [5] Sander Dieleman. *"Noise Schedules Considered Harmful."* [Link](https://sander.ai/2024/06/14/noise-schedules). $ $ ### Conclusion In summary, the effectiveness of **MoNL** for **multimodal diffusion models**, particularly with **audio** and **video** data, stems from its strategic use of **connection** and **heterogeneity**. By applying **variable noise levels**, MoNL enhances **connectivity** between modalities and better adapts to diverse **temporal** and **frequency components**, leading to superior performance compared to existing multimodal learning methods.
Follow-Up on Theoretical Clarifications and Request for Final Review
Dear Reviewer SA7V, Thank you for your feedback and for raising your score after reviewing our rebuttal. We appreciate your acknowledgment of our efforts to clarify the experimental aspects of our work. In response to your concern about the theoretical intuition, we want to highlight that we have newly provided the **"Theoretical Background on Mixture of Noise Levels,"** which we hope you will consider as part of our ongoing effort to strengthen our work and contribute to the field. As we have only two days left for further discussion, we kindly request that you review our additional responses. We sincerely appreciate the time and effort you have dedicated to reviewing our paper and your constructive and insightful comments. Thank you once again. Best regards, The Authors
Summary
This paper introduces the Audiovisual Diffusion Transformer (AVDiT) with Mixture of Noise Levels (MoNL) for audiovisual sequence generation. The key innovation is the use of variable noise levels during the diffusion process, applied across different time segments and modalities. This approach enables the model to effectively learn arbitrary conditional distributions in a task-agnostic manner, making it versatile for various generation tasks such as cross-modal generation, multimodal interpolation, and audiovisual continuation. Experiments on multiple datasets demonstrate that AVDiT with MoNL outperforms existing baselines, generating temporally and perceptually consistent audiovisual sequences.
Strengths
1. The proposed variable noise levels across different time segments and modalities is an effective approach to enhance the flexibility of diffusion models. 2. The model's ability to handle various audiovisual generation tasks within a single framework is impressive. 3. The paper includes extensive experiments on multiple datasets, with both qualitative and quantitative evaluations. The additional demo page provides an intuitive comparison.
Weaknesses
1. While the variable noise levels concept is interesting, the overall novelty of the approach may be seen as incremental. Similar techniques in diffusion models and transformers have been explored in papers like MM-Diffusion. 2. Some technical details are not thoroughly explained, such as the criteria for selecting and varying noise levels across time segments and modalities. 3. The evaluation is primarily conducted on a few datasets (Monologues, AIST++, Landscape). The model's generalizability to real-world scenarios remains uncertain. 4. The authors compare the transformer-based AVDiT model to the UNet-based MM-Diffusion model, which might not be a fair comparison due to the different architectures. The authors should consider training a UNet-based model using their proposed approach to provide a more direct comparison and validate the effectiveness of their method.
Questions
See weakness.
Rating
5
Confidence
4
Soundness
3
Presentation
2
Contribution
3
Limitations
The authors mention some limitations in the appendix, such as the need for further improvements in visual and speech quality and the potential for ethical concerns. However, for me, the lack of extensive evaluation of more diverse and real-world datasets is a noteworthy limitation. Some additional limitations can be found in the Weaknesses section.
Theoretical Background on Mixture of Noise Levels
## Theoretical Background on Mixture of Noise Levels $ $ ### 1. Theoretical Background on Multimodal Learning In *"A Theory of Multimodal Learning"* [1], multimodal learning is shown to offer a superior **generalization bound** compared to unimodal learning, with an improvement factor of **$O(\sqrt{n})$**, where **$n$** denotes the sample size. This benefit relies on **connection** and **heterogeneity** between modalities: - **Connection**: The bound depends on learned connections between (**$\mathcal{X}$**) and (**$\mathcal{Y}$**). - **Heterogeneity**: Describes how modalities, **$\mathcal{X}$** and **$\mathcal{Y}$**, diverge and complement. If connection and heterogeneity are missing, ill-conditioned scenarios can arise. For instance, if **$x \equiv y$**, perfect connection suggests no need for learning about **$\mathcal{Y}$**. On the other hand, if **$x$** is random noise, there is heterogeneity but no meaningful connection between **$\mathcal{X}$** and **$\mathcal{Y}$**, making non-trivial learning on **$\mathcal{X}$** alone impractical. The theory also highlights that and learning effective connections between modalities via **generative models** can enhance multimodal learning. This forms the basis for our **Mixture of Noise Levels (MoNL)** approach, which is particularly suited for multimodal learning with **audio** and **video** data. [1] Zhou Lu. *"A Theory of Multimodal Learning."* NeurIPS 2023. $ $ ### 2. Advantages of Mixture of Noise Level Training (MoNL) Our **Mixture of Noise Level (MoNL)** training method offers significant benefits for multimodal learning, especially with **audio** and **video** data: - **Heterogeneity and Connection**: Audio and video are naturally heterogeneous. For example, a video of a person speaking includes **audio** of spoken words and **video** of lip movements and facial expressions. MoNL uses **variable noise levels** to enhance learning by capturing the **generic transition matrix** across the **temporal axis**. $$p _{\mathbf{\theta}}([\mathbf{z} _{t ^{(1,1)}-1} ^{(1,1)}, \ldots, \mathbf{z} _{t ^{(M, N)}-1} ^{(M, N)}] \mid[\mathbf{z} _{t ^{(1,1)}} ^{(1,1)}, \ldots, \mathbf{z} _{t ^{(M, N)}} ^{(M, N)}]) \qquad \text{(Eq. (4))}$$ where $M$, $N$ are the number of modalities and time-segments, respectively. - **Enhanced Connectivity**: MoNL improves **connectivity** between **audio** and **video** modalities. Our experiments show that MoNL often surpasses **task-specific learning** approaches by fostering better connections between modalities, adapting its focus more effectively. $ $ ### 3. Enhanced Connectivity - Comparison with Existing Methods - **MoNL vs. Joint Learning in MMD** [2]: Unlike joint learning methods that focus on the joint distribution $p_{\mathbf{\theta}}(\mathbf{z} _{t-1} \mid\mathbf{z} _{t})$, MoNL trains across **multiple conditioning**, enabling better connections by varying its focus. This is evidenced by MoNL outperforming the Vanilla (see Table 1) and MMD models (see Tables 2 and 3). - **MoNL vs. Per-Modality Training**: MoNL goes beyond per-modality training in UniDiffuser [3], which uses variable noise between modalities i.e., learning $p_{\mathbf{\theta}}([\mathbf{z} _{t ^{(1)}-1} ^{(1)}, \ldots, \mathbf{z} _{t ^{(M)}-1} ^{(M)}] \mid[\mathbf{z} _{t ^{(1)}} ^{(1)}, \ldots, \mathbf{z} _{t ^{(M)}} ^{(M)}])$. MoNL introduces variable noise across different **time segments**, learning connections across **temporal dynamics** as well. This advantage is demonstrated in Table 1. - **MoNL vs. Masked Training** [4]: Diffusion models often obscure high-frequency details with low noise and low-frequency structures with high noise [5]. MoNL employs variable noise levels to explore diverse **frequency components**, enhancing the model's ability to correlate high and low-frequency elements. This is in contrast to masked self-supervised learning, which limits frequency-specific connections by masking entire elements. [2] Ruan et al. *"MM-Diffusion: Learning Multi-Modal Diffusion Models for Joint Audio and Video Generation."* CVPR 2023. [3] Bao et al. *"One Transformer Fits All Distributions in Multi-Modal Diffusion at Scale."* ICML 2023. [4] Voleti et al. *"MCVD: Masked Conditional Video Diffusion for Prediction, Generation, and Interpolation."* NeurIPS 2022. [5] Sander Dieleman. *"Noise Schedules Considered Harmful."* [Link](https://sander.ai/2024/06/14/noise-schedules). $ $ ### Conclusion In summary, the effectiveness of **MoNL** for **multimodal diffusion models**, particularly with **audio** and **video** data, stems from its strategic use of **connection** and **heterogeneity**. By applying **variable noise levels**, MoNL enhances **connectivity** between modalities and better adapts to diverse **temporal** and **frequency components**, leading to superior performance compared to existing multimodal learning methods.
Request for Final Review: Response and Theoretical Clarifications
Dear Reviewer tkoE, We kindly request that you review our responses, as we have only **two days** left for further **discussion**. We have addressed your comments thoughtfully and provided a thorough **theoretical background** of our method. We sincerely appreciate the time and effort you have dedicated to reviewing our paper and your constructive and insightful feedback. Thank you once again. Best regards, The Authors
Thank you for addressing all my comments. Overall, it is a good paper on audiovisual generation, although I find the proposed method to be somewhat incremental. I suggest the authors make their trained models and code available for reproducibility. I'm willing to increase my rating to BA.
Thank you for your time and effort
Dear Reviewer tkoE, We sincerely appreciate your time and effort in reviewing our paper. We are glad that you found our work to be a good contribution to the field of audiovisual generation. We believe that our AVDiT with MoNL effectively models temporal dependencies, handles diverse multimodal tasks, and demonstrates superior performance compared to existing approaches, offering significant advancements beyond prior art. We believe these contributions can be valuable to the community! We are committed to fostering reproducibility in our research and are happy to make our trained models and code publicly available upon acceptance of the paper. Thank you again for your valuable insights. Sincerely, The Authors
Summary
The paper presents a new method for audiovisual generation where the input output condiitions may comprise of 2 modalities, namely video and audio sequence. The authors propose a new training approach to effectively learn conditional distribution in multimodal space. The main novely in the paper is a mixture of noise level formulation for processing the inputs. The method produces temporally consistent samples and outperforms existing arts and the vanilla configuration
Strengths
1. The utilization of mixture of noise levels is novel and the methods seems to improve robustness in denoising, hence leading to a performance boost 2. The evaluations seems to be fair and clearly signifies the working of the method. 3. The paper is well written and easy to follow.
Weaknesses
1. Although the method seems to work well empirically, the paper lacks theoretical backing for the proposed method. It would be good to see some proofs that the proposed method leads to a better approximation of the variational lower bound and joint distribution. 2. Are there any sampling modifications required to accommodate the proposed training strategy? 3. There is a plethora of compositional works derived from the energy based formulation of diffusion models. Could the authors analyze how the proposed method performs in comparison to it. [1] https://energy-based-model.github.io/Compositional-Visual-Generation-with-Composable-Diffusion-Models/ 4. I'm rating the paper borderline for now, I will improve my rating if the authors can address my concerns
Questions
1. Could the authors provide a proof of why the proposed method would work better 2. Are there any sampling modifications required to accommodate the proposed training strategy? 3. Could the authors give an analysis of how the proposed algorithm will work when compared to [1]
Rating
6
Confidence
3
Soundness
2
Presentation
3
Contribution
2
Limitations
Please see weakness.
Theoretical Background on Mixture of Noise Levels
## Theoretical Background on Mixture of Noise Levels $ $ ### 1. Theoretical Background on Multimodal Learning In *"A Theory of Multimodal Learning"* [1], multimodal learning is shown to offer a superior **generalization bound** compared to unimodal learning, with an improvement factor of **$O(\sqrt{n})$**, where **$n$** denotes the sample size. This benefit relies on **connection** and **heterogeneity** between modalities: - **Connection**: The bound depends on learned connections between (**$\mathcal{X}$**) and (**$\mathcal{Y}$**). - **Heterogeneity**: Describes how modalities, **$\mathcal{X}$** and **$\mathcal{Y}$**, diverge and complement. If connection and heterogeneity are missing, ill-conditioned scenarios can arise. For instance, if **$x \equiv y$**, perfect connection suggests no need for learning about **$\mathcal{Y}$**. On the other hand, if **$x$** is random noise, there is heterogeneity but no meaningful connection between **$\mathcal{X}$** and **$\mathcal{Y}$**, making non-trivial learning on **$\mathcal{X}$** alone impractical. The theory also highlights that and learning effective connections between modalities via **generative models** can enhance multimodal learning. This forms the basis for our **Mixture of Noise Levels (MoNL)** approach, which is particularly suited for multimodal learning with **audio** and **video** data. [1] Zhou Lu. *"A Theory of Multimodal Learning."* NeurIPS 2023. $ $ ### 2. Advantages of Mixture of Noise Level Training (MoNL) Our **Mixture of Noise Level (MoNL)** training method offers significant benefits for multimodal learning, especially with **audio** and **video** data: - **Heterogeneity and Connection**: Audio and video are naturally heterogeneous. For example, a video of a person speaking includes **audio** of spoken words and **video** of lip movements and facial expressions. MoNL uses **variable noise levels** to enhance learning by capturing the **generic transition matrix** across the **temporal axis**. $$p _{\mathbf{\theta}}([\mathbf{z} _{t ^{(1,1)}-1} ^{(1,1)}, \ldots, \mathbf{z} _{t ^{(M, N)}-1} ^{(M, N)}] \mid[\mathbf{z} _{t ^{(1,1)}} ^{(1,1)}, \ldots, \mathbf{z} _{t ^{(M, N)}} ^{(M, N)}]) \qquad \text{(Eq. (4))}$$ where $M$, $N$ are the number of modalities and time-segments, respectively. - **Enhanced Connectivity**: MoNL improves **connectivity** between **audio** and **video** modalities. Our experiments show that MoNL often surpasses **task-specific learning** approaches by fostering better connections between modalities, adapting its focus more effectively. $ $ ### 3. Enhanced Connectivity - Comparison with Existing Methods - **MoNL vs. Joint Learning in MMD** [2]: Unlike joint learning methods that focus on the joint distribution $p_{\mathbf{\theta}}(\mathbf{z} _{t-1} \mid\mathbf{z} _{t})$, MoNL trains across **multiple conditioning**, enabling better connections by varying its focus. This is evidenced by MoNL outperforming the Vanilla (see Table 1) and MMD models (see Tables 2 and 3). - **MoNL vs. Per-Modality Training**: MoNL goes beyond per-modality training in UniDiffuser [3], which uses variable noise between modalities i.e., learning $p_{\mathbf{\theta}}([\mathbf{z} _{t ^{(1)}-1} ^{(1)}, \ldots, \mathbf{z} _{t ^{(M)}-1} ^{(M)}] \mid[\mathbf{z} _{t ^{(1)}} ^{(1)}, \ldots, \mathbf{z} _{t ^{(M)}} ^{(M)}])$. MoNL introduces variable noise across different **time segments**, learning connections across **temporal dynamics** as well. This advantage is demonstrated in Table 1. - **MoNL vs. Masked Training** [4]: Diffusion models often obscure high-frequency details with low noise and low-frequency structures with high noise [5]. MoNL employs variable noise levels to explore diverse **frequency components**, enhancing the model's ability to correlate high and low-frequency elements. This is in contrast to masked self-supervised learning, which limits frequency-specific connections by masking entire elements. [2] Ruan et al. *"MM-Diffusion: Learning Multi-Modal Diffusion Models for Joint Audio and Video Generation."* CVPR 2023. [3] Bao et al. *"One Transformer Fits All Distributions in Multi-Modal Diffusion at Scale."* ICML 2023. [4] Voleti et al. *"MCVD: Masked Conditional Video Diffusion for Prediction, Generation, and Interpolation."* NeurIPS 2022. [5] Sander Dieleman. *"Noise Schedules Considered Harmful."* [Link](https://sander.ai/2024/06/14/noise-schedules). $ $ ### Conclusion In summary, the effectiveness of **MoNL** for **multimodal diffusion models**, particularly with **audio** and **video** data, stems from its strategic use of **connection** and **heterogeneity**. By applying **variable noise levels**, MoNL enhances **connectivity** between modalities and better adapts to diverse **temporal** and **frequency components**, leading to superior performance compared to existing multimodal learning methods.
Request for Final Review: Response and Theoretical Clarifications
Dear Reviewer bNvz, We kindly request that you review our responses, as we have only **two days** left for further **discussion**. We have addressed your comments thoughtfully and provided a thorough **theoretical background** of our method. We sincerely appreciate the time and effort you have dedicated to reviewing our paper and your constructive and insightful feedback. Thank you once again. Best regards, The Authors
Dear Reviewer bNvz, We would like to gently remind you that we have **only one day remaining** for further discussion on our manuscript. We have carefully considered your valuable feedback and provided **theoritical backing** for our methodology. We greatly appreciate your time and insights thus far. Thank you again for your contributions. Sincerely, The Authors
Dear Authors, I thank you for the detailed explanations and clarifications. After careful consideration and going through the other reviews I’m increasing my rating to weak Accept. The reason of not giving a higher score is due to lack of theoretical novelty.
Dear Reviewer bNvz, We sincerely appreciate your time and thoughtful feedback. We are grateful for your recognition of the detailed explanations and clarifications provided, and for upgrading our paper to a "weak accept." We recognize the need for stronger theoretical foundations. While we acknowledge the current limitations in the theoretical landscape of multimodal learning as highlighted in *"A Theory of Multimodal Learning."* (Zhou et al. NeurIPS 2023), we believe our research makes a substantial empirical contribution to this field. Our findings serve as a robust foundation for future theoretical explorations and advancements. We are committed to addressing the theoretical aspects of our work in greater depth in our ongoing research. Thank you once again for your valuable insights. Sincerely, The Authors
Thanks for the response
Thanks for the response. I have read the response as well as the other reviews. I have considered the author's rebuttal, but it does not provide new or compelling information that would change my evaluation. Therefore, I will maintain my current score.
Post-rebuttal
I appreciate the author's efforts in preparing the rebuttal. After reading the rebuttal, I think some of my concerns related to the experiments are clarified. On the other hand, this is a rather empirical paper, and I cannot always find the underlying theoretical intuition of this entire line of work. However, maybe the brute-force scaling up can overpass the underlying theoretical justification over the challenge of learning disjoint distributions without much understanding of the data distribution and it seems to work fine. Anyway, I raise my score to 5 for post-rebuttal.
Request for Final Review: Response and Theoretical Clarifications
Dear Reviewers, We kindly request that you review our responses, as we have only **two days** left for further **discussion**. We have addressed your comments thoughtfully and provided a thorough **theoretical background** of our method, which will be incorporated in the final version. We sincerely appreciate the time and effort you have dedicated to reviewing our paper and your constructive and insightful feedback. Thank you once again. Best regards, The Authors
Decision
Accept (poster)