arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.11316cs.CVcs.LG

Transformer Module Networks for Systematic Generalization in Visual Question Answering

Fujitsu Limited(富士通有限公司);Fujitsu Research of America, Inc.(美国富士通研究公司);Massachusetts Institute of Technology(麻省理工学院);Center for Brains, Minds and Machines(大脑、心智与机器中心)

Moyuru Yamada, Vanessa D'Amario, Kentaro Takemoto, Xavier Boix, Tomotake Sasaki

英文摘要:

Transformers achieve great performance on Visual Question Answering (VQA). However, their systematic generalization capabilities, i.e., handling novel combinations of known concepts, is unclear. We reveal that Neural Module Networks (NMNs), i.e., question-specific compositions of modules that tackle a sub-task, achieve better or similar systematic generalization performance than the conventional Transformers, even though NMNs' modules are CNN-based. In order to address this shortcoming of Transformers with respect to NMNs, in this paper we investigate whether and how modularity can bring benefits to Transformers. Namely, we introduce Transformer Module Network (TMN), a novel NMN based on compositions of Transformer modules. TMNs achieve state-of-the-art systematic generalization performance in three VQA datasets, improving more than 30% over standard Transformers for novel compositions of sub-tasks. We show that not only the module composition but also the module specialization for each sub-task are the key of such performance gain.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑