Generalizable Image Repair for Robust Visual Control

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Sobolewski, Carson, Mao, Zhenjiang, Vejre, Kshitij Maruti, Ruchkin, Ivan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911085126221824
author Sobolewski, Carson
Mao, Zhenjiang
Vejre, Kshitij Maruti
Ruchkin, Ivan
author_facet Sobolewski, Carson
Mao, Zhenjiang
Vejre, Kshitij Maruti
Ruchkin, Ivan
contents Vision-based control relies on accurate perception to achieve robustness. However, image distribution changes caused by sensor noise, adverse weather, and dynamic lighting can degrade perception, leading to suboptimal control decisions. Existing approaches, including domain adaptation and adversarial training, improve robustness but struggle to generalize to unseen corruptions while introducing computational overhead. To address this challenge, we propose a real-time image repair module that restores corrupted images before they are used by the controller. Our method leverages generative adversarial models, specifically CycleGAN and pix2pix, for image repair. CycleGAN enables unpaired image-to-image translation to adapt to novel corruptions, while pix2pix exploits paired image data when available to improve the quality. To ensure alignment with control performance, we introduce a control-focused loss function that prioritizes perceptual consistency in repaired images. We evaluated our method in a simulated autonomous racing environment with various visual corruptions. The results show that our approach significantly improves performance compared to baselines, mitigating distribution shift and enhancing controller reliability.
format Preprint
id arxiv_https___arxiv_org_abs_2503_05911
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Generalizable Image Repair for Robust Visual Control
Sobolewski, Carson
Mao, Zhenjiang
Vejre, Kshitij Maruti
Ruchkin, Ivan
Robotics
Computer Vision and Pattern Recognition
Vision-based control relies on accurate perception to achieve robustness. However, image distribution changes caused by sensor noise, adverse weather, and dynamic lighting can degrade perception, leading to suboptimal control decisions. Existing approaches, including domain adaptation and adversarial training, improve robustness but struggle to generalize to unseen corruptions while introducing computational overhead. To address this challenge, we propose a real-time image repair module that restores corrupted images before they are used by the controller. Our method leverages generative adversarial models, specifically CycleGAN and pix2pix, for image repair. CycleGAN enables unpaired image-to-image translation to adapt to novel corruptions, while pix2pix exploits paired image data when available to improve the quality. To ensure alignment with control performance, we introduce a control-focused loss function that prioritizes perceptual consistency in repaired images. We evaluated our method in a simulated autonomous racing environment with various visual corruptions. The results show that our approach significantly improves performance compared to baselines, mitigating distribution shift and enhancing controller reliability.
title Generalizable Image Repair for Robust Visual Control
topic Robotics
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.05911