arXiv:2510.10068cs.CV2025-10被引 1

用随机掩码自动编码器构建概率超图,实现多模态多任务半监督学习。

Probabilistic Hyper-Graphs using Multiple Randomly Masked Autoencoders for Semi-supervised Multi-modal Multi-task Learning

  • 通过整体模态随机掩码生成超边分布,构建概率超图
  • 单训练循环融合预训练与微调,提升下游任务性能
  • 支持推理时集成,且可轻量知识蒸馏,适合小模型应用

计算机视觉领域得益于多模态数据的丰富性,在各类视觉任务中取得显著进展。近年来,掩码自动编码器(MAE)作为自监督预训练方法备受关注,通常在下游任务(如分类或回归)前使用,无需人工标注数据。本文提出基于掩码自动编码器的概率超图(PHG-MAE):将经典神经图模型与现代MAE方法统一于共同理论框架下。通过随机掩码整个模态而非仅图像块,模型在每次前向传播中采样超边分布。此外,该模型改进标准MAE,将预训练与微调整合至单一训练循环。同时,支持推理时集成,通过聚合提升预测性能与一致性。最后,我们证明可在集成基础上进行知识蒸馏,即使参数少于100万的模型也几乎无性能损失。研究聚焦于包含多种世界解释的户外无人机场景,但方法可推广至自动驾驶、室内机器人等类似领域。为简化外部预训练专家的集成流程,我们开发了数据流水线工具,并自动化扩展了Dronescapes数据集。所有技术细节、代码及复现步骤均已公开。

原文摘要 · Abstract (English)

The computer vision domain has greatly benefited from an abundance of data across many modalities to improve on various visual tasks. Recently, there has been a lot of focus on self-supervised pre-training methods through Masked Autoencoders (MAE) \cite{he2022masked,bachmann2022multimae}, usually used as a first step before optimizing for a downstream task, such as classification or regression. This is very useful as it doesn't require any manually labeled data. In this work, we introduce Probabilistic Hyper-Graphs using Masked Autoencoders (PHG-MAE): a novel model that unifies the classical work on neural graphs \cite{leordeanu2021semi} with the modern approach of masked autoencoders under a common theoretical framework. Through random masking of entire modalities, not just patches, the model samples from the distribution of hyper-edges on each forward pass. Additionally, the model adapts the standard MAE algorithm by combining pre-training and fine-tuning into a single training loop. Moreover, our approach enables the creation of inference-time ensembles which, through aggregation, boost the final prediction performance and consistency. Lastly, we show that we can apply knowledge distillation on top of the ensembles with little loss in performance, even with models that have fewer than 1M parameters. While our work mostly focuses on outdoor UAV scenes that contain multiple world interpretations and modalities, the same steps can be followed in other similar domains, such as autonomous driving or indoor robotics. In order to streamline the process of integrating external pre-trained experts for computer vision multi-modal multi-task learning (MTL) scenarios, we developed a data-pipeline software. Using this tool, we have created and released a fully-automated extension of the Dronescapes dataset. All the technical details, code and reproduction steps are publicly released.

多模态自监督超图集成学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。