← Search

Zedong Wang

12 accepted papers

2026

CARE-Edit: Condition-Aware Routing of Experts for Contextual Image Editing

CVPR 2026

Unified diffusion editors often rely on a fixed, shared backbone for diverse tasks, suffering from task interference and poor adaptation to heterogeneous demands (e.g., local vs global, semantic vs photometric). In particular, prevalent ControlNet and OmniControl variants combine multiple conditioni

Cited by 0SourcecodeScholar
2026

Steady-State Behavior of Constant-Stepsize Stochastic Approximation: Gaussian Approximation and Tail Bounds

ICML 2026poster

Constant-stepsize stochastic approximation is widely used in learning for computational efficiency. For a fixed stepsize, the iterates typically admit a stationary distribution that is rarely tractable. Prior work shows that as the stepsize $\alpha \downarrow 0$, the centered-and-scaled steady state…

Cited by 0SourceScholar
2025

MergeVQ: A Unified Framework for Visual Generation and Representation with Disentangled Token Merging and Quantization

CVPR 2025poster

Masked Image Modeling (MIM) with Vector Quantization (VQ) has achieved great success in both self-supervised pre-training and image generation. However, most existing methods struggle to address the trade-off in the shared latent space for generation quality vs. representation learning and efficienc…

2025

Prior-guided Hierarchical Harmonization Network for Efficient Image Dehazing

AAAI 2025technical

Image dehazing is a crucial task that involves the enhancement of degraded images to recover their sharpness and textures. While vision Transformers have exhibited impressive results in diverse dehazing tasks, their quadratic complexity and lack of dehazing priors pose significant drawbacks for real…

Cited by 0SourcePDFScholar
2025

Rep-MTL: Unleashing the Power of Representation-level Task Saliency for Multi-Task Learning

ICCV 2025poster

Despite the promise of Multi-Task Learning (MTL) in leveraging complementary knowledge across tasks, existing multi-task optimization (MTO) techniques remain fixated on resolving conflicts through optimizer-centric loss scaling and gradient manipulation, yet fail to deliver consistent gains. In this…

Cited by 0SourcePDFScholar
2025

Taming LLMs with Gradient Grouping

ACL 2025long

Training large language models (LLMs) poses challenges due to their massive scale and heterogeneous architectures. While adaptive optimizers like AdamW help address gradient variations, they still struggle with efficient and effective parameter-wise learning rate estimation, resulting in training in…

2024

LongVQ: Long Sequence Modeling with Vector Quantization on Structured Memory

IJCAI 2024poster

Transformer models have been successful in various sequence processing tasks, but the self-attention mechanism's computational cost limits its practicality for long sequences. Although there are existing attention variants that improve computational efficiency, they have a limited ability to abstrac…

Cited by 2SourcePDFScholar
2024

MogaNet: Multi-order Gated Aggregation Network

ICLR 2024poster

By contextualizing the kernel as global as possible, Modern ConvNets have shown great potential in computer vision tasks. However, recent progress on \textit{multi-order game-theoretic interaction} within deep neural networks (DNNs) reveals the representation bottleneck of modern ConvNets, where the…

2024

SemiReward: A General Reward Model for Semi-supervised Learning

ICLR 2024poster

Semi-supervised learning (SSL) has witnessed great progress with various improvements in the self-training framework with pseudo labeling. The main challenge is how to distinguish high-quality pseudo labels against the confirmation bias. However, existing pseudo-label selection strategies are limite…

2024

Short-Long Convolutions Help Hardware-Efficient Linear Attention to Focus on Long Sequences

ICML 2024poster

To mitigate the computational complexity in the self-attention mechanism on long sequences, linear attention utilizes computation tricks to achieve linear complexity, while state space models (SSMs) popularize a favourable practice of using non-data-dependent memory pattern, *i.e.,* emphasize the ne…

Cited by 6SourcePDFScholar
2024

VQDNA: Unleashing the Power of Vector Quantization for Multi-Species Genomic Sequence Modeling

ICML 2024poster

Similar to natural language models, pre-trained genome language models are proposed to capture the underlying intricacies within genomes with unsupervised sequence modeling. They have become essential tools for researchers and practitioners in biology. However, the hand-crafted tokenization policies…

Cited by 9SourcePDFScholar
2023

OpenSTL: A Comprehensive Benchmark of Spatio-Temporal Predictive Learning

NeurIPS 2023poster

Spatio-temporal predictive learning is a learning paradigm that enables models to learn spatial and temporal patterns by predicting future frames from given past frames in an unsupervised manner. Despite remarkable progress in recent years, a lack of systematic understanding persists due to the dive…