MATA: A Trainable Hierarchical Automaton System for Multi-Agent Visual Reasoning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cai, Zhixi, Ke, Fucai, Leo, Kevin, Huang, Sukai, de la Banda, Maria Garcia, Stuckey, Peter J., Rezatofighi, Hamid
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911402104455168
author Cai, Zhixi
Ke, Fucai
Leo, Kevin
Huang, Sukai
de la Banda, Maria Garcia
Stuckey, Peter J.
Rezatofighi, Hamid
author_facet Cai, Zhixi
Ke, Fucai
Leo, Kevin
Huang, Sukai
de la Banda, Maria Garcia
Stuckey, Peter J.
Rezatofighi, Hamid
contents Recent vision-language models have strong perceptual ability but their implicit reasoning is hard to explain and easily generates hallucinations on complex queries. Compositional methods improve interpretability, but most rely on a single agent or hand-crafted pipeline and cannot decide when to collaborate across complementary agents or compete among overlapping ones. We introduce MATA (Multi-Agent hierarchical Trainable Automaton), a multi-agent system presented as a hierarchical finite-state automaton for visual reasoning whose top-level transitions are chosen by a trainable hyper agent. Each agent corresponds to a state in the hyper automaton, and runs a small rule-based sub-automaton for reliable micro-control. All agents read and write a shared memory, yielding transparent execution history. To supervise the hyper agent's transition policy, we build transition-trajectory trees and transform to memory-to-next-state pairs, forming the MATA-SFT-90K dataset for supervised finetuning (SFT). The finetuned LLM as the transition policy understands the query and the capacity of agents, and it can efficiently choose the optimal agent to solve the task. Across multiple visual reasoning benchmarks, MATA achieves the state-of-the-art results compared with monolithic and compositional baselines. The code and dataset are available at https://github.com/ControlNet/MATA.
format Preprint
id arxiv_https___arxiv_org_abs_2601_19204
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle MATA: A Trainable Hierarchical Automaton System for Multi-Agent Visual Reasoning
Cai, Zhixi
Ke, Fucai
Leo, Kevin
Huang, Sukai
de la Banda, Maria Garcia
Stuckey, Peter J.
Rezatofighi, Hamid
Artificial Intelligence
Computer Vision and Pattern Recognition
Recent vision-language models have strong perceptual ability but their implicit reasoning is hard to explain and easily generates hallucinations on complex queries. Compositional methods improve interpretability, but most rely on a single agent or hand-crafted pipeline and cannot decide when to collaborate across complementary agents or compete among overlapping ones. We introduce MATA (Multi-Agent hierarchical Trainable Automaton), a multi-agent system presented as a hierarchical finite-state automaton for visual reasoning whose top-level transitions are chosen by a trainable hyper agent. Each agent corresponds to a state in the hyper automaton, and runs a small rule-based sub-automaton for reliable micro-control. All agents read and write a shared memory, yielding transparent execution history. To supervise the hyper agent's transition policy, we build transition-trajectory trees and transform to memory-to-next-state pairs, forming the MATA-SFT-90K dataset for supervised finetuning (SFT). The finetuned LLM as the transition policy understands the query and the capacity of agents, and it can efficiently choose the optimal agent to solve the task. Across multiple visual reasoning benchmarks, MATA achieves the state-of-the-art results compared with monolithic and compositional baselines. The code and dataset are available at https://github.com/ControlNet/MATA.
title MATA: A Trainable Hierarchical Automaton System for Multi-Agent Visual Reasoning
topic Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2601.19204