arXiv · 2302.08242
Tuning computer vision models with task rewards
Abstract
Misalignment between model predictions and intended usage can be detrimental for the deployment of computer vision models. The issue is exacerbated when the task involves complex structured outputs, as it becomes harder to design procedures which address this misalignment. In natural language processing, this is often addressed using reinforcement learning techniques that align models with a task reward. We adopt this approach and show its surprising effectiveness across multiple computer vision tasks, such as object detection, panoptic segmentation, colorization and image captioning. We believe this approach has the potential to be widely useful for better aligning models with a diverse range of computer vision tasks.
Explore related subjects
Keep this discovery
André Susano Pinto, Alexander Kolesnikov, Yuge Shi, Lucas Beyer, Xiaohua Zhai. 2023-02-16. Tuning computer vision models with task rewards. https://arxiv.org/abs/2302.08242
Cite the original work for its findings. Save a collection to share your selection of sources.