Tuning computer vision models with task rewards
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 36 papers in the map (data through 7 Oct 2026). Builds on 37 of its 48 references that are in the map. First posted on arXiv 16 Feb 2023.
Builds on
37 of 48 references.
References · 37
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2015Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networkscited by 9,617
- 2020End-to-End Object Detection with Transformerscited by 7,384
- 2021Zero-Shot Text-to-Image Generationcited by 4,276
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 2020Learning to summarize from human feedbackcited by 2,562
- 2014CIDEr: Consensus-based Image Description Evaluationcited by 2,027
- 2021Scaling Vision Transformerscited by 993
25 more
- 2014Deep Visual-Semantic Alignments for Generating Image Descriptionscited by 966
- 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 896
- 2018Panoptic Segmentationcited by 796
- 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 779
- 2015Sequence Level Training with Recurrent Neural Networkscited by 568
- 2021How to train your ViT? Data, Augmentation, and Regularization in Vision Transformerscited by 562
- 2016Self-critical Sequence Training for Image Captioningcited by 544
- 2022Improving alignment of dialogue agents via targeted human judgementscited by 527
- 2021Pix2seq: A Language Modeling Framework for Object Detectioncited by 345
- 2019Generalization in Generation: A closer look at Exposure Biascited by 123
- 2015Minimum Risk Training for Neural Machine Translationcited by 114
- 2019On NMT Search Errors and Model Errors: Cat Got Your Tongue?cited by 114
- 2022UViM: A Unified Modeling Approach for Vision with Learned Guiding Codescited by 77
- 2018Can Neural Machine Translation be Improved with User Feedback?cited by 71
- 2021Machine Translation Decoding beyond Beam Searchcited by 55
- 2021Deep Reinforcement Learning in Computer Vision: A Comprehensive Surveycited by 42
- 2018Deep Reinforcement Learning For Sequence to Sequence Modelscited by 39
- 2019Reinforced Feature Points: Optimizing Feature Detection and Description for a High-Level Taskcited by 39
- 2022End-to-End Transformer Based Model for Image Captioningcited by 37
- 2016End-to-end training of object class detectors for mean average precisioncited by 35
- 2015Training Deep Neural Networks via Direct Loss Minimizationcited by 25
- 2022Exploiting Multiple Sequence Lengths in Fast End to End Training for Image Captioningcited by 18
- 2016PoseAgent: Budget-Constrained 6D Object Pose Estimation via Reinforcement Learningcited by 9
- 2021MetricOpt: Learning to Optimize Black-Box Evaluation Metricscited by 9
- 2014Closed-Form Training of Conditional Random Fields for Large Scale Image Segmentationcited by 1
Built on by
Citing papers per year: 2023: 15; 2024: 11; 2025: 5; 2026: 5.
36 papers.
Sort:
By area:
- 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewardscited by 232
- 2023HIVE: Harnessing Human Feedback for Instructional Visual Editingcited by 172
- 20234M: Massively Multimodal Masked Modelingcited by 118
- 2023LLMScore: Unveiling the Power of Large Language Models in Text-to-Image Synthesis Evaluationcited by 85
- 2024BOND: Aligning LLMs with Best-of-N Distillationcited by 73
- 2023Large AI Models in Health Informatics: Applications, Challenges, and the Futurecited by 61
- 2023Test-Time Adaptation with CLIP Reward for Zero-Shot Generalization in Vision-Language Modelscited by 37
- 2024LocCa: Visual Pretraining with Location-aware Captionerscited by 35
- 2024ReinDiffuse: Crafting Physically Plausible Motions with Reinforced Diffusion Modelcited by 25
- 2023Carve3D: Improving Multi-view Reconstruction Consistency for Diffusion Models with RL Finetuningcited by 22
- 2023Reinforced UI Instruction Grounding: Towards a Generic UI Task Automation APIcited by 18
- 2023UnifiedGesture: A Unified Gesture Synthesis Model for Multiple Skeletonscited by 15
24 more
- 2023Cross-Domain Image Captioning with Discriminative Finetuningcited by 14
- 2023Reward Finetuning for Faster and More Accurate Unsupervised Object Discoverycited by 14
- 2025SpatialThinker: Reinforcing Scene Graph-Grounded Spatial Reasoning via Dense Rewardscited by 14
- 2023TaskMet: Task-Driven Metric Learning for Model Learningcited by 11
- 2025From Screens to Scenes: A Survey of Embodied AI in Healthcarecited by 10
- 2024Reinforcement Learning Gradients as Vitamin for Online Finetuning Decision Transformerscited by 8
- 2025Graph-Augmented Reasoning: Evolving Step-by-Step Knowledge Graph Retrieval for LLM Reasoningcited by 8
- 2023Uncertainty Estimation for Safety-critical Scene Segmentation via Fine-grained Reward Maximizationcited by 7
- 2025Large (Vision) Language Models are Unsupervised In-Context Learnerscited by 6
- 2023An Invitation to Deep Reinforcement Learningcited by 5
- 2024Lane2Seq: Towards Unified Lane Detection via Sequence Generationcited by 5
- 2024No Detail Left Behind: Revisiting Self-Retrieval for Fine-Grained Image Captioningcited by 5
- 2024RoboKoop: Efficient Control Conditioned Representations from Visual Input in Robotics using Koopman Operatorcited by 4
- 2024Aligning Vision Models with Human Aesthetics in Retrieval: Benchmarks and Algorithmscited by 3
- 2025Test-Time Adaptation with Binary Feedbackcited by 3
- 2024Optimizing Novelty of Top-k Recommendations using Large Language Models and Reinforcement Learningcited by 2
- 2024Evolutionary Retrofittingcited by 2
- 2026EchoAvatar: Real-time Generative Avatar Animation from Audio Streamscited by 1
- 2023A dynamical clipping approach with task feedback for Proximal Policy Optimizationcited by 0
- 2024Incorporating System-level Safety Requirements in Perception Models via Reinforcement Learningcited by 0
- 2026Policy Gradient with Adaptive Entropy Annealing for Continual Fine-Tuningcited by 0
- 2026On the Feasibility and Opportunity of Autoregressive 3D Object Detectioncited by 0
- 2026Falcon Perception-HD: High Density Perception via Reinforcement Learningcited by 0
- 2026LookThere! Sparse Vision by Reinforced Selectioncited by 0
- 2026LookThere! Sparse Vision by Reinforced Selection4 Sep 2026
- 2026Falcon Perception-HD: High Density Perception via Reinforcement Learning19 Aug 2026
- 2026EchoAvatar: Real-time Generative Avatar Animation from Audio Streams27 May 2026
- 2026On the Feasibility and Opportunity of Autoregressive 3D Object Detection9 Mar 2026
- 2026Policy Gradient with Adaptive Entropy Annealing for Continual Fine-Tuning15 Feb 2026
- 2025SpatialThinker: Reinforcing Scene Graph-Grounded Spatial Reasoning via Dense Rewards10 Nov 2025
- 2025Test-Time Adaptation with Binary Feedback24 May 2025
- 2025Large (Vision) Language Models are Unsupervised In-Context Learners3 Apr 2025
- 2025Graph-Augmented Reasoning: Evolving Step-by-Step Knowledge Graph Retrieval for LLM Reasoning3 Mar 2025
- 2025From Screens to Scenes: A Survey of Embodied AI in Healthcare13 Jan 2025
- 2024Incorporating System-level Safety Requirements in Perception Models via Reinforcement Learning4 Dec 2024
- 2024Reinforcement Learning Gradients as Vitamin for Online Finetuning Decision Transformers31 Oct 2024
24 more
- 2024Evolutionary Retrofitting15 Oct 2024
- 2024ReinDiffuse: Crafting Physically Plausible Motions with Reinforced Diffusion Model9 Oct 2024
- 2024RoboKoop: Efficient Control Conditioned Representations from Visual Input in Robotics using Koopman Operator4 Sep 2024
- 2024No Detail Left Behind: Revisiting Self-Retrieval for Fine-Grained Image Captioning4 Sep 2024
- 2024BOND: Aligning LLMs with Best-of-N Distillation19 Jul 2024
- 2024Optimizing Novelty of Top-k Recommendations using Large Language Models and Reinforcement Learning20 Jun 2024
- 2024Aligning Vision Models with Human Aesthetics in Retrieval: Benchmarks and Algorithms13 Jun 2024
- 2024LocCa: Visual Pretraining with Location-aware Captioners28 Mar 2024
- 2024Lane2Seq: Towards Unified Lane Detection via Sequence Generation27 Feb 2024
- 2023Carve3D: Improving Multi-view Reconstruction Consistency for Diffusion Models with RL Finetuning21 Dec 2023
- 2023An Invitation to Deep Reinforcement Learning13 Dec 2023
- 2023A dynamical clipping approach with task feedback for Proximal Policy Optimization12 Dec 2023
- 20234M: Massively Multimodal Masked Modeling11 Dec 2023
- 2023TaskMet: Task-Driven Metric Learning for Model Learning8 Dec 2023
- 2023Uncertainty Estimation for Safety-critical Scene Segmentation via Fine-grained Reward Maximization5 Nov 2023
- 2023Reward Finetuning for Faster and More Accurate Unsupervised Object Discovery29 Oct 2023
- 2023Reinforced UI Instruction Grounding: Towards a Generic UI Task Automation API7 Oct 2023
- 2023UnifiedGesture: A Unified Gesture Synthesis Model for Multiple Skeletons13 Sep 2023
- 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewards7 Jun 2023
- 2023Test-Time Adaptation with CLIP Reward for Zero-Shot Generalization in Vision-Language Models29 May 2023
- 2023LLMScore: Unveiling the Power of Large Language Models in Text-to-Image Synthesis Evaluation18 May 2023
- 2023Cross-Domain Image Captioning with Discriminative Finetuning4 Apr 2023
- 2023Large AI Models in Health Informatics: Applications, Challenges, and the Future21 Mar 2023
- 2023HIVE: Harnessing Human Feedback for Instructional Visual Editing16 Mar 2023
- 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewardscited by 232
- 2023HIVE: Harnessing Human Feedback for Instructional Visual Editingcited by 172
- 2024BOND: Aligning LLMs with Best-of-N Distillationcited by 73
- 2023Reward Finetuning for Faster and More Accurate Unsupervised Object Discoverycited by 14
- 2024Aligning Vision Models with Human Aesthetics in Retrieval: Benchmarks and Algorithmscited by 3
- 2024BOND: Aligning LLMs with Best-of-N Distillation19 Jul 2024
- 2024Aligning Vision Models with Human Aesthetics in Retrieval: Benchmarks and Algorithms13 Jun 2024
- 2023Reward Finetuning for Faster and More Accurate Unsupervised Object Discovery29 Oct 2023
- 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewards7 Jun 2023
- 2023HIVE: Harnessing Human Feedback for Instructional Visual Editing16 Mar 2023
- 20234M: Massively Multimodal Masked Modelingcited by 118
- 2023LLMScore: Unveiling the Power of Large Language Models in Text-to-Image Synthesis Evaluationcited by 85
- 2023Large AI Models in Health Informatics: Applications, Challenges, and the Futurecited by 61
- 2024LocCa: Visual Pretraining with Location-aware Captionerscited by 35
- 2023Reinforced UI Instruction Grounding: Towards a Generic UI Task Automation APIcited by 18
- 2023Cross-Domain Image Captioning with Discriminative Finetuningcited by 14
- 2023TaskMet: Task-Driven Metric Learning for Model Learningcited by 11
- 2025From Screens to Scenes: A Survey of Embodied AI in Healthcarecited by 10
- 2023Uncertainty Estimation for Safety-critical Scene Segmentation via Fine-grained Reward Maximizationcited by 7
- 2023An Invitation to Deep Reinforcement Learningcited by 5
- 2024Lane2Seq: Towards Unified Lane Detection via Sequence Generationcited by 5
- 2024No Detail Left Behind: Revisiting Self-Retrieval for Fine-Grained Image Captioningcited by 5
12 most cited of 23 All 23 (JSON)
- 2026LookThere! Sparse Vision by Reinforced Selection4 Sep 2026
- 2026Falcon Perception-HD: High Density Perception via Reinforcement Learning19 Aug 2026
- 2026EchoAvatar: Real-time Generative Avatar Animation from Audio Streams27 May 2026
- 2026On the Feasibility and Opportunity of Autoregressive 3D Object Detection9 Mar 2026
- 2026Policy Gradient with Adaptive Entropy Annealing for Continual Fine-Tuning15 Feb 2026
- 2025Test-Time Adaptation with Binary Feedback24 May 2025
- 2025From Screens to Scenes: A Survey of Embodied AI in Healthcare13 Jan 2025
- 2024Incorporating System-level Safety Requirements in Perception Models via Reinforcement Learning4 Dec 2024
- 2024Evolutionary Retrofitting15 Oct 2024
- 2024RoboKoop: Efficient Control Conditioned Representations from Visual Input in Robotics using Koopman Operator4 Sep 2024
- 2024No Detail Left Behind: Revisiting Self-Retrieval for Fine-Grained Image Captioning4 Sep 2024
- 2024Optimizing Novelty of Top-k Recommendations using Large Language Models and Reinforcement Learning20 Jun 2024
12 newest of 23 All 23 (JSON)