Dual Mean-Teacher: An Unbiased Semi-Supervised Framework for Audio-Visual Source Localization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Guo, Yuxin, Ma, Shijie, Su, Hu, Wang, Zhiqing, Zhao, Yuhao, Zou, Wei, Sun, Siyang, Zheng, Yun
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916147602915328
author Guo, Yuxin
Ma, Shijie
Su, Hu
Wang, Zhiqing
Zhao, Yuhao
Zou, Wei
Sun, Siyang
Zheng, Yun
author_facet Guo, Yuxin
Ma, Shijie
Su, Hu
Wang, Zhiqing
Zhao, Yuhao
Zou, Wei
Sun, Siyang
Zheng, Yun
contents Audio-Visual Source Localization (AVSL) aims to locate sounding objects within video frames given the paired audio clips. Existing methods predominantly rely on self-supervised contrastive learning of audio-visual correspondence. Without any bounding-box annotations, they struggle to achieve precise localization, especially for small objects, and suffer from blurry boundaries and false positives. Moreover, the naive semi-supervised method is poor in fully leveraging the information of abundant unlabeled data. In this paper, we propose a novel semi-supervised learning framework for AVSL, namely Dual Mean-Teacher (DMT), comprising two teacher-student structures to circumvent the confirmation bias issue. Specifically, two teachers, pre-trained on limited labeled data, are employed to filter out noisy samples via the consensus between their predictions, and then generate high-quality pseudo-labels by intersecting their confidence maps. The sufficient utilization of both labeled and unlabeled data and the proposed unbiased framework enable DMT to outperform current state-of-the-art methods by a large margin, with CIoU of 90.4% and 48.8% on Flickr-SoundNet and VGG-Sound Source, obtaining 8.9%, 9.6% and 4.6%, 6.4% improvements over self- and semi-supervised methods respectively, given only 3% positional-annotations. We also extend our framework to some existing AVSL methods and consistently boost their performance.
format Preprint
id arxiv_https___arxiv_org_abs_2403_03145
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Dual Mean-Teacher: An Unbiased Semi-Supervised Framework for Audio-Visual Source Localization
Guo, Yuxin
Ma, Shijie
Su, Hu
Wang, Zhiqing
Zhao, Yuhao
Zou, Wei
Sun, Siyang
Zheng, Yun
Computer Vision and Pattern Recognition
Machine Learning
Multimedia
Sound
Audio and Speech Processing
Audio-Visual Source Localization (AVSL) aims to locate sounding objects within video frames given the paired audio clips. Existing methods predominantly rely on self-supervised contrastive learning of audio-visual correspondence. Without any bounding-box annotations, they struggle to achieve precise localization, especially for small objects, and suffer from blurry boundaries and false positives. Moreover, the naive semi-supervised method is poor in fully leveraging the information of abundant unlabeled data. In this paper, we propose a novel semi-supervised learning framework for AVSL, namely Dual Mean-Teacher (DMT), comprising two teacher-student structures to circumvent the confirmation bias issue. Specifically, two teachers, pre-trained on limited labeled data, are employed to filter out noisy samples via the consensus between their predictions, and then generate high-quality pseudo-labels by intersecting their confidence maps. The sufficient utilization of both labeled and unlabeled data and the proposed unbiased framework enable DMT to outperform current state-of-the-art methods by a large margin, with CIoU of 90.4% and 48.8% on Flickr-SoundNet and VGG-Sound Source, obtaining 8.9%, 9.6% and 4.6%, 6.4% improvements over self- and semi-supervised methods respectively, given only 3% positional-annotations. We also extend our framework to some existing AVSL methods and consistently boost their performance.
title Dual Mean-Teacher: An Unbiased Semi-Supervised Framework for Audio-Visual Source Localization
topic Computer Vision and Pattern Recognition
Machine Learning
Multimedia
Sound
Audio and Speech Processing
url https://arxiv.org/abs/2403.03145