Degradation Identification Agent Ai
The overall framework of Ai. Ai can evaluate all types of degradation levels in an all-in-one framework. It can process videos, along with prompts, to identify the degradations. It consists of a vision encoder to extract both spatial and temporal features and a text tokenizer to tokenize the input prompts. These features are projected into the same space by trained projectors. A pre-trained LLM is utilized to fuse the features while fine-tuned with LoRA.
Routing and Restoration Agent Ar
Illumination of degradation removal process by Routing and Restoration Agent Ar. Ar is able to route the degradation removal orders, rollback when restoration fails, reroute to another degradation removal orders.
Quality Assessment Agent Aa
An overview of quality assessment agent Aa. It consists of three feature encoders, including an image feature extractor for extracting spatial features from sparse video frames, a motion feature extractor for extracting motion features from the entire video, and a text encoder for extracting aligned text features from prompts. The extracted features are then aligned through projectors and fed into a pre-trained LLM to generate the output results. LoRA weights are introduced to the pre-trained image encoder and the large language model to adapt the models to the quality assessment task.
Agent Collaboration and Closed-Loop Design
MoA-VR incorporates three specialized agents within a closed-loop architecture. For a low-quality input video, Ai identifies the degradation type and level; Ar generates a degradation removal plan and then invokes the corresponding restoration toolbox; Aa assesses all the intermediate results and chooses the best quality one. Then Ai identifies whether the previous restoration was successful. If it fails, Ar rolls back and reroutes; if successful, Ar follows the previous plan. This loop continues until all degradations are removed.
Performance on restoration task
Performance comparison between state-of-the-art all-in-one methods.
Performance of different agents
Evaluation of degradation identification agent, routing and restoration agent and quality assessment agent..