Sketching without Worrying: Noise-Tolerant Sketch-Based Image Retrieval
Ayan Kumar BhuniaSubhadeep KoleyAbdullah Faiz Ur Rahman KhiljiAneeshan SainPinaki Nath ChowdhuryTao XiangYi-Zhe Song
Proposes a reinforcement learning-based stroke subset selector that filters out noisy or detrimental strokes from amateur drawings to boost fine-grained sketch-based image retrieval accuracy without retraining underlying retrieval models.
Sketch-based image retrieval enables users to search for specific photos by drawing on interactive touchscreens. However, widespread adoption faces a major hurdle termed the fear-to-sketch barrier, where amateur users lack confidence in their drawing ability. The article investigates this issue and discovers that retrieval failures are rarely caused by an inability to sketch basic outlines. Instead, failures stem from irrelevant or inaccurate strokes that act as noise and severely degrade search accuracy. The main objective of the article is to demonstrate an intelligent preprocessing module that detects and removes these noisy strokes before image retrieval, enabling high-precision visual search from imperfect freehand drawings.
To achieve this, the authors designed a stroke subset selector operating directly on sequential sketch coordinates. The system pairs a hierarchical recurrent neural network, which models relationships between individual strokes and the overall drawing, with a reinforcement learning training scheme. Because drawing coordinates must be converted into raster images to query standard retrieval models, the process cannot use traditional derivative-based training. The selector is trained using an actor-critic algorithm where a fixed retrieval model serves as the critic, rewarding the selector whenever it picks a stroke combination that improves retrieval accuracy. The authors evaluated this framework on standard benchmark datasets comprising thousands of shoe and chair sketches.
Experimental results show that the proposed selector substantially improves search performance across all settings. When added to standard retrieval models, top-ranked retrieval accuracy increased by approximately 8% to 10%, achieving 43.7% accuracy on shoes and 64.8% on chairs, outperforming existing state-of-the-art systems. In simulated extreme conditions with heavy synthetic noise, the selector restored retrieval accuracy from 13.4% back up to 37.2%. Additionally, the evaluation score produced by the critic network reliably measures whether an incomplete sketch has enough information to search effectively, allowing systems to reduce unnecessary visual processing by 42.2% during interactive drawing.
These findings demonstrate that overcoming user drawing limitations does not require building massive, complicated retrieval architectures. Instead, lightweight preprocessing in the stroke coordinate domain offers a highly effective, plug-and-play solution. This approach improves system robustness and reduces user frustration while maintaining low computational overhead, adding only about 18.3% extra central processing unit time and roughly 22.4% more mathematical operations compared to baseline retrieval methods.
Organizations developing interactive visual search or sketch-based applications should adopt stroke subset selection as a standard front-end filter. Teams can also utilize the trained selector as a data augmentation tool to generate clean partial sketches for downstream training, avoiding unstable alternative methods. Future work should focus on extending this framework beyond simple product categories to multi-object complex scenes, as current testing was confined to isolated single-object benchmark datasets. Overall, the findings provide strong confidence that noisy stroke elimination is a practical and scalable solution for real-world sketch applications.
- Paper: DRAW: A Recurrent Neural Network For Image Generation, Karol Gregor et al. (2015). Introduces the foundational recurrent neural network framework for sequential, iterative stroke-based visual generation and attention that underpins coordinate-level stroke modeling.
- Paper: Learning Fine-Grained Image Similarity with Deep Ranking, Jiang Wang et al. (2014). Establishes the deep metric learning and ranking architectures essential for fine-grained image retrieval systems used as downstream evaluators.
- Paper: Learning Robust Global Representations by Penalizing Local Predictive Power, Haohan Wang et al. (2019). Provides fundamental insights and benchmark data on overcoming local superficial noise to learn robust global shape representations from sketches.
- Paper: Sketch-based manga retrieval using manga109 dataset, Yusuke Matsui et al. (2015). Presents foundational techniques for line-art feature extraction and nearest-neighbor matching in sketch-based visual retrieval pipelines.
- Paper: Picture that Sketch: Photorealistic Image Generation from Abstract Sketches, Subhadeep Koley et al. (2023). Extends the handling of amateur, abstract sketch imperfections from retrieval filtering to high-fidelity photorealistic image generation.
- Paper: Breathing Life Into Sketches Using Text-to-Video Priors, Rinon Gal et al. (2024). Builds on parametric stroke-level representations to animate abstract freehand sketches without raster distortion using generative video priors.
- Paper: Composed Image Retrieval with Text Feedback via Multi-grained Uncertainty Regularization, Yiyang Chen et al. (2024). Explores uncertainty regularization to handle ambiguity and coarse-to-fine user query refinement in interactive image retrieval.
- Paper: SVGDreamer: Text Guided SVG Generation with Diffusion Model, Ximing Xing et al. (2024). Applies vector-stroke coordinate optimization to generate editable, clean multi-path vector graphics from abstract prompts.
