Task-Oriented Data Synthesis and Control-Rectify Sampling for Remote Sensing Semantic Segmentation
On the map: Diffusion models → Inverse problems and image restoration
Abstract (arXiv)
With the rapid progress of controllable generation, training data synthesis has become a promising way to expand labeled datasets and alleviate manual annotation in remote sensing (RS). However, the complexity of semantic mask control and the uncertainty of sampling quality often limit the utility of synthetic data in downstream semantic segmentation tasks. To address these challenges, we propose a task-oriented data synthesis framework (TODSynth), including a Multimodal Diffusion Transformer (MM-DiT) with unified triple attention and a plug-and-play sampling strategy guided by task feedback. Built upon the powerful DiT-based generative foundation model, we systematically evaluate different control schemes, showing that a text-image-mask joint attention scheme combined with full fine-tuning of the image and mask branches significantly enhances the effectiveness of RS semantic segmentation data synthesis, particularly in few-shot and complex-scene scenarios. Furthermore, we propose a control-rectify flow matching (CRFM) method, which dynamically adjusts sampling directions guided by semantic loss during the early high-plasticity stage, mitigating the instability of generated images and bridging the gap between synthetic data and downstream segmentation tasks. Extensive experiments demonstrate that our approach consistently outperforms state-of-the-art controllable generation methods, producing more stable and task-oriented synthetic data for RS semantic segmentation.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Inverse problems and image restoration, one of 14 directions in Diffusion models. Builds on 47 of its 59 references that are in the map. First posted on arXiv 18 Dec 2025.
Builds on
47 of 59 references.
Closest ancestors · 5
Same direction, newest first
- 2025Object Fidelity Diffusion for Remote Sensing Image Generation14 Aug
- 2025EarthSynth: Generating Informative Earth Observation with Diffusion Models17 May
- 2025DiT4SR: Taming Diffusion Transformer for Real-World Image Super-Resolution30 Mar
- 2024SatSynth: Augmenting Image-Mask Pairs through Diffusion Models for Aerial Semantic Segmentation25 Mar
- 2023DiffusionSat: A Generative Foundation Model for Satellite Imagery6 Dec
Other references · 42
- 2022Scalable Diffusion Models with Transformerscited by 5,931
- 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,466
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,290
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2016Pyramid Scene Parsing Networkcited by 2,472
- 2021Masked-attention Mask Transformer for Universal Image Segmentationcited by 2,338
- 2023T2I-Adapter: Learning Adapters to Dig out More Controllable Ability for Text-to-Image Diffusion Modelscited by 1,339
- 2021Segmenter: Transformer for Semantic Segmentationcited by 855
- 2023DataComp: In search of the next generation of multimodal datasetscited by 635
- 2022Dataset Distillation by Matching Training Trajectoriescited by 437
30 more
- 2023Uni-ControlNet: All-in-One Control to Text-to-Image Diffusion Modelscited by 358
- 2024OminiControl: Minimal and Universal Control for Diffusion Transformercited by 334
- 2024Hunyuan-DiT: A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understandingcited by 294
- 2021LoveDA: A Remote Sensing Land-Cover Dataset for Domain Adaptive Semantic Segmentationcited by 259
- 2023LayoutDiffusion: Controllable Diffusion Model for Layout-to-image Generationcited by 258
- 2023OmniControl: Control Any Joint at Any Time for Human Motion Generationcited by 233
- 2024FlowEdit: Inversion-Free Text-Based Editing Using Pre-Trained Flow Modelscited by 214
- 2022Dataset Condensation via Efficient Synthetic-Data Parameterizationcited by 187
- 2022Scaling Up Dataset Distillation to ImageNet-1K with Constant Memorycited by 180
- 2024ControlNeXt: Powerful and Efficient Control for Image and Video Generationcited by 158
- 2025Less-to-More Generalization: Unlocking More Controllability by In-Context Generationcited by 153
- 2022Minimizing the Accumulated Trajectory Error to Improve Dataset Distillationcited by 142
- 2024Lumina-T2X: Transforming Text into Any Modality, Resolution, and Duration via Flow-based Large Diffusion Transformerscited by 137
- 2018Learning Semantic Segmentation from Synthetic Data: A Geometrically Guided Input-Output Adaptation Approachcited by 119
- 2025EasyControl: Adding Efficient and Flexible Control for Diffusion Transformercited by 118
- 2023Dataset Diffusion: Diffusion-based Synthetic Dataset Generation for Pixel-Level Semantic Segmentationcited by 112
- 2022Dataset Condensation with Contrastive Signalscited by 105
- 2023DataDAM: Efficient Dataset Distillation with Attention Matchingcited by 83
- 2023Freestyle Layout-to-Image Synthesiscited by 67
- 2023FreeMask: Synthetic Images with Dense Annotations Make Stronger Segmentation Modelscited by 61
- 2024CreatiLayout: Siamese Multimodal Diffusion Transformer for Creative Layout-to-Image Generationcited by 58
- 2025Dataset Distillation with Neural Characteristic Function: A Minmax Perspectivecited by 45
- 2024EmerDiff: Emerging Pixel-level Semantic Knowledge in Diffusion Modelscited by 42
- 2024AeroGen: Enhancing Remote Sensing Object Detection with Diffusion-Driven Data Generationcited by 32
- 2025UniCombine: Unified Multi-Conditional Combination with Diffusion Transformercited by 25
- 2025SplitFlow: Flow Decomposition for Inversion-Free Text-to-Image Editingcited by 15
- 2024FUSU: A Multi-temporal-source Land Use Change Segmentation Dataset for Fine-grained Urban Semantic Understandingcited by 11
- 2025A Training-free Synthetic Data Selection Method for Semantic Segmentationcited by 9
- 2024Training-Free Layout-to-Image Generation with Marginal Attention Constraintscited by 4
- 2025TerraGen: A Unified Multi-Task Layout Generation Framework for Remote Sensing Data Augmentationcited by 4
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025A Gift from the Integration of Discriminative and Diffusion-based Generative Learning: Boundary Refinement Remote Sensing Semantic Segmentation
- 2024P-MSDiff: Parallel Multi-Scale Diffusion for Remote Sensing Image Segmentation
- 2026Mitigating Long-Tail Bias via Prompt-Controlled Diffusion Augmentation
- 2025DiffRIS: Enhancing Referring Remote Sensing Image Segmentation with Pre-trained Text-to-Image Diffusion Models
- 2026Towards Realistic Remote Sensing Dataset Distillation with Discriminative Prototype-guided Diffusion
- 2025RSRWKV: A Linear-Complexity 2D Attention Mechanism for Efficient Remote Sensing Vision Task
- 2026SHARP: Spectrum-aware Highly-dynamic Adaptation for Resolution Promotion in Remote Sensing Synthesis
- 2026VPRef: A Cross-Domain Benchmark for Referring Remote Sensing Image Segmentation