Ryan Ko | arXiv Science

arXiv2026

ReVision : A Post-Hoc, Vision-Based Technique for Replacing Unacceptable Concepts in Image Generation Pipeline

Image-generative models are widely deployed across industries. Recent studies show that they can be exploited to produce unacceptable content. Existing mitigation strategies rely on prompt filtering and safety-aware training, both of which can be bypassed and often degrade generative quality. In this work, we propose ReVision, a training-free, prompt-based, post-hoc safety framework for image-generation pipeline. ReVision acts as a post-generation safeguard by analyzing generated images and selectively editing unsafe concepts without altering the underlying generator. Prior post-hoc editing methods often rely on imprecise spatial localization, limiting deployability, in multi-concept scenes. To address this limitation, ReVision introduces a VLM-assisted spatial gating mechanism for instance-consistent localization, enabling integrity-preserving edits. We introduce an 800-image benchmark spanning single- and multi-unsafe-concept images, each composed alongside benign concepts in shared scenes. On this benchmark, ReVision improves CLIP alignment toward safe prompts by +0.121, reduces multi-concept background LPIPS from 0.166 to 0.058, and eliminates NudeNet detections (70.51 -> 0). Across external benchmarks, ReVision outperforms prior methods, and a human study shows it reduces recognizability of unacceptable content from 96% to 10%.

cs.CR↗

arXiv2021

FDGATII : Fast Dynamic Graph Attention with Initial Residual and Identity Mapping

While Graph Neural Networks have gained popularity in multiple domains, graph-structured input remains a major challenge due to (a) over-smoothing, (b) noisy neighbours (heterophily), and (c) the suspended animation problem. To address all these problems simultaneously, we propose a novel graph neural network FDGATII, inspired by attention mechanism's ability to focus on selective information supplemented with two feature preserving mechanisms. FDGATII combines Initial Residuals and Identity Mapping with the more expressive dynamic self-attention to handle noise prevalent from the neighbourhoods in heterophilic data sets. By using sparse dynamic attention, FDGATII is inherently parallelizable in design, whist efficient in operation; thus theoretically able to scale to arbitrary graphs with ease. Our approach has been extensively evaluated on 7 datasets. We show that FDGATII outperforms GAT and GCN based benchmarks in accuracy and performance on fully supervised tasks, obtaining state-of-the-art results on Chameleon and Cornell datasets with zero domain-specific graph pre-processing, and demonstrate its versatility and fairness.

cs.LG↗