Search arXiv⌕ Search

arXiv subjects

Jianguo Li

Publications and source records attributed to Jianguo Li.

At least 19 recordsLinked to original sources

Intrinsic Distribution Matching

One-step and few-step text-to-image generation must accurately render the details specified in the prompt despite limited opportunities for refinement. Distribution matching distillation (DMD) uses score differences to bring the generated distribution closer to a reference distribution. Following prior work, we estimate this difference with internal velocity branches, avoiding a separate fake-score network. We ask whether an explicit moment constraint improves this learned score supervision. A decomposition of the reverse-KL field into a Gaussian moment component and a non-Gaussian residual identifies statistics that can be constrained explicitly. We introduce Intrinsic Distribution Matching (IDM), which retains the full DMD correction and matches feature means and per-coordinate standard deviations using the generator's exponential-moving-average (EMA) copy, requiring no external feature encoder. Experiments use full-parameter and low-rank adaptation (LoRA) tuning on Z-Image-Turbo 6B and Qwen-Image 20B. At two network evaluations, IDM reaches 94.25% and 90.20% on Chinese and English LongText-Bench, the highest scores among the compared methods at this budget. Controlled ablations show gains from the added moment loss, while generator EMA features perform competitively with an external DINOv2 encoder across the five reported metrics. Code is available at https://github.com/inclusionAI/IDM.

cs.CV↗

UBTree: Parallel Tree Drafting via Unigram and Bigram Models for Speculative Decoding

Speculative decoding accelerates language model inference by verifying multiple draft tokens in a single target-model pass. Recent parallel drafters have achieved breakthrough performance in frontier production models, but their effectiveness deteriorates as the entropy of target distributions increases due to insufficient draft diversity. To overcome this bottleneck without sacrificing parallelism, we introduce UBTree, a parallel drafter that couples a Unigram proposer with a Bigram selector to construct drafting Trees. The unigram proposer is trained with the standard cross-entropy objective to generate candidate tokens independently for each position, while a lightweight bigram selector predicts transition scores between adjacent candidate pairs. Unlike the proposer, the selector is trained with a renormalized KL objective on high-temperature data. This tree-native training broadens the supervision beyond the greedy path, encouraging plausible alternative branches that improve the chance of accepting additional tokens during tree verification. Across seven standardized benchmarks with Qwen3-4B and Qwen3-8B, UBTree achieves an average speedup of $5.84$--$6.94\times$ over autoregressive decoding and outperforms DARTree in all 28 comparisons. Production-scale evaluation further demonstrates UBTree's advantage over frontier baselines such as DSpark.

cs.CL↗

UpLoop: Upcycling Large Language Models with Gated Loop

Looped large language models trade parameter storage for recurrent computation, but most are trained from scratch. In practice, pretraining with looped depth is expensive, making post-training upcycling of a pretrained non-looped LLM an attractive alternative. We introduce \textbf{UpLoop}, a post-training recipe based on \emph{layer-wise} recurrence. Instead of looping a whole model or a middle block, UpLoop loops each selected layer locally, preserving the pretrained depth hierarchy and avoiding feature-space mismatch. To make loop execution useful, we add hidden-wise gates on the attention and MLP residual branches, so later loops act as progressive refinements rather than redundant rewrites. Empirically, UpLoop breaks the commonly observed saturation at two loops ($R=2$) in prior looped models. Starting from Qwen3-4B Base and applying SFT, UpLoop improves the average accuracy across 20 benchmarks by 6.17\% at $R=3$. We use $R=3$ as the main setting for the best accuracy--cost trade-off, while both native and extrapolated $R=4$ yield further gains.

cs.LG↗

Widely Linear-Enhanced IQ Imbalance Interference Cancellation for AFDM-FTN Systems

Internet of Things (IoT) networks require reliable and spectrally efficient wireless links to support massive device connectivity and high mobility applications. To meet these requirements, this paper investigates a widely linear (WL)-empowered receiver design for affine frequency division multiplexing (AFDM) with faster-than-Nyquist (FTN) signaling under practical receiver in phase and quadrature imbalance (IQI). The considered AFDM-FTN system couples the direct and image branches with FTN induced intersymbol interference (FTN-ISI) and delay-Doppler spreading in the affine domain. To address this coupled interference structure, a unified conjugate augmented input output model is established for IQI impaired AFDM-FTN systems. Based on this model, a WL minimum mean square error receiver with parallel interference cancellation (WL-MMSE-PIC) is developed, where WL-MMSE preprocessing is followed by the proposed hard decision PIC (HD-PIC) and damped soft decision PIC (SD-PIC). The proposed receiver is further analyzed in terms of the local stability of HD-PIC, a genie aided performance reference under ideal residual interference cancellation, and the impact of the AFDM chirp parameter on the post equalization interference distribution. Simulation results show that the proposed WL-MMSE-HD-PIC and WL-MMSE-SD-PIC receivers achieve significant error rate performance gains over the conventional linear-MMSE and WL-MMSE baselines. The results further show that FTN signaling provides a nominal spectral-efficiency gain, while residual-coupling-based chirp-parameter selection reduces the post-equalization residual coupling and improves detection reliability for high-mobility IoT links.

eess.SP↗

A Low-Complexity Message-Passing Receiver for FTN-WAFT under Nonlinear Distortion

This letter proposes a faster-than-Nyquist (FTN) weighted affine Fourier transform (WAFT) transmission scheme, termed FTN-WAFT, which combines the peak-to-average power ratio (PAPR) flexibility of WAFT with FTN time compression to improve spectral efficiency and power amplifier (PA) efficiency over high-mobility channels. However, operating the PA at a reduced input back-off (IBO) inevitably introduces nonlinear distortion, which further couples with FTN-induced inter-symbol interference (FTN-ISI) and doubly selective channel dispersion. To address this practical impairment, a multilayer message-passing (MLMP) receiver aided by PA interference cancellation (PA-IC), termed PA-IC-MLMP, is proposed. The proposed receiver reconstructs the dominant PA distortion from soft symbol estimates, cancels the channel-propagated nonlinear interference, and performs MLMP over the WAFT, FTN, channel, and constellation layers. Simulation results show that the proposed receiver improves the bit error rate (BER) and effective throughput under nonlinear PA constraints while maintaining low complexity.

eess.SP↗

Reducing Pretraining-Generation Mismatch in Diffusion Language Models

Diffusion language models (dLLMs) generate text through iterative denoising, allowing multiple tokens to be predicted in parallel. However, pretraining may mask tokens throughout a sequence, whereas prompt continuation conditions on an intact prefix. This difference remains in conversion pipelines that denoise entire sequences during the stable stage. We propose Prefix-Conditioned Diffusion (PCD), which samples a boundary, preserves the prefix, and denoises the suffix. The training recipe also applies autoregressive supervision to the prefix. We evaluate PCD in the stable stage of a warmup, stable, and decay conversion pipeline, with inference unchanged. Matched experiments across model families show improvements in reasoning and coding over native diffusion training. A matched continuation study further shows that the advantage persists after a shared decay stage. In a separate reconstruction diagnostic, the full PCD recipe's advantage over native diffusion training reverses as more evaluation prefix tokens are masked. Controlled experiments show lower suffix reconstruction loss with a clean training prefix, an intact evaluation prefix, and no autoregressive loss.

cs.CL↗

An Unbounded Archive-based Transfer Strategy for Dynamic Multi-Objective Optimization with a Changing Number of Objectives

Dynamic multi-objective optimization with a variable number of objectives is difficult because objective-dimensional variations may significantly change the Pareto front and degrade algorithm adaptability. This paper proposes an unbounded archive-based transfer strategy (UATS), which maintains an unbounded archive of offspring solutions within each environment stage and extracts feasible nondominated solutions as transferable elites when objective changes occur. UATS is embedded into SPEA2SDE to construct UATS-SPEA2SDE, enabling the algorithm to reuse historical evolutionary information while retaining the convergence and diversity advantages of shift-based density estimation. Experiments are conducted on four benchmark problems under three objective-changing settings, where UATS-SPEA2SDE is compared with a restart-based SPEA2SDE baseline and four representative dynamic multi-objective optimization algorithms. The results indicate that the archive-guided transfer improves recovery after environmental changes and enhances adaptability to objective-number variations.

cs.NE↗

LLaDA-UI: Bringing Block-wise Diffusion to Vision-Language GUI Agents

Diffusion large language models (dLLMs) achieve high decoding efficiency through block-parallel, arbitrary-order generation, making them attractive for latency-sensitive applications. GUI agents represent a natural testbed for this paradigm, as they must repeatedly perceive screen states and emit structured, spatially grounded actions in real time. However, whether dLLMs can be extended into capable multimodal GUI agents while preserving their parallel decoding advantage remains an open question. We present LLaDA-UI, a 16.7B-parameter MoE-based, block-wise diffusion vision-language GUI agent. LLaDA-UI follows a two-stage training pipeline: general multimodal pre-training aligns a native-resolution vision encoder with the LLaDA2.0-mini-base diffusion language backbone, followed by GUI-agent supervised fine-tuning on diverse mobile, desktop, web, and grounding data. Across widely adopted grounding benchmarks and navigation benchmarks spanning multiple platforms, LLaDA-UI substantially outperforms Qwen2.5-VL-7B and surpasses Qwen3-VL-8B on four of six reported GUI benchmarks. These results establish block-wise diffusion as a practical generative paradigm for multimodal GUI agents.

cs.CV↗

LLaDA-Image: Building Strong Image Generators with Fully Open Training Recipes

We introduce LLaDA-Image, a unified framework that pairs a 6B Diffusion Transformer (DiT) trained from scratch with a frozen vision-language understanding module built on the LLaDA2.0-Mini diffusion language model backbone. Instead of relying heavily on paired image-text data from the beginning, we first build a strong visual generative prior through image-only pre-training and mid-training. The generation pipeline comprises 220M samples, 98 of which are real images. For efficient and scalable optimization, we use parameter-free RMSNorm throughout the DiT together with the Muon optimizer. The resulting unified model produces highly photorealistic images while accurately following fine-grained editing instructions. We further distill LLaDA-Image into LLaDA-Image-Turbo, enabling fast inference in 2-4 sampling steps. On Qwen-Image-Bench, LLaDA-Image achieves overall scores of 53.53 and 53.38 on the English and Chinese tracks, respectively, setting a new state-of-the-art among open-source models on both tracks. To support further research on capable and efficient generative models, we release our model weights, training code, and detailed recipes.

cs.CV↗

UI-Venus-2 Technical Report

Multimodal GUI agents have emerged as a promising paradigm for digital task automation, yet transitioning from benchmark-oriented models to dependable real-world applications remains challenging due to limited environment coverage, brittle task construction, and unreliable reward verification. In this work, we present UI-Venus-2, a general-purpose foundation GUI agent designed to operate across mobile, web, and desktop environments through a unified closed-loop reasoning-action framework. To bridge the gap toward practical deployment, we jointly scale three critical dimensions: (1) Environments, expanding coverage to more than 170 multilingual mobile apps and native desktop operating systems; (2) Tasks, employing a deep-research pipeline for function-grounded instruction generation; and (3) Verification, adopting trace-level and sample-level evaluators with visual keypoints and multi-model voting to ensure reliable RL signals for training. Furthermore, we integrate safety-aware mechanisms to ensure controlled execution of consequential actions. By offering a capable, efficient, and open-source foundation, UI-Venus-2 advances the field toward more generalizable, verifiable, and self-reflective agents for real-world applications.

cs.AI↗

Resource-Efficient Bio-Molecular Docking on a NISQ-era Digital Quantum Computer

Molecular docking is a vital computational task in drug discovery, wherein the objective is to efficiently identify optimal binding poses between a ligand and a target receptor protein. Due to the combinatorial explosion of possible binding configurations, docking of large and flexible molecules remains a computationally intensive problem, especially at scale. Early studies have revealed that the molecular docking can be re-cast as a maximum vertex-weighted clique problem (MVWCP) problem on a compatibility graph to be solved classically. In this work, we proposed a hybrid quantum-classical approach for molecular docking leveraging the MVWCP formalism with a variational full-basis encoding (FBE) strategy, which enables efficient encoding of classical binary variables with Bloch sphere vectors. We further prove that a global minimizer of the FBE objective can always be chosen to be a pure product state, thereby providing a rigorous justification for its optimization using a unitary variational circuit. The molecular docking problem is first mapped to a cost Hamiltonian that is minimized within a variational framework, optimized via a randomized imaginary time evolution (ITE)-inspired warm start, and gradient-based techniques. Finally, we also executed the circuit on an IBM quantum computer, underlying the feasibility and of quantum-assisted optimization for structure-based drug design and point towards the broader utility of advanced encoding techniques in quantum optimization for computational biology.

quant-ph↗

Unlocking the Potential of Image Editing via Concept Scaling and Dense Supervision

Existing image editing frameworks predominantly follow the training paradigm of text-to-image diffusion models. However, extending this paradigm to image editing highlights two inherent discrepancies, specifically, the insufficient attention to edit concept granularity and the training inefficiency caused by sparse supervision signals. To address these issues, we establish a comprehensive hierarchical taxonomy featuring over 1,000 fine-grained edit concepts and build ConceptEdit-12M, a massive dataset of 12 million high-quality editing pairs via an improved synthesis framework. This library-driven approach effectively rectifies the distribution collapse of generated data while ensuring high data fidelity. Furthermore, we propose a dense supervision training strategy that synthesizes multiple non-interfering concepts into single image pairs. By providing richer learning signals, this strategy significantly enhances both training efficiency and overall model performance. Training results validate our strategy, significantly outperforming prior works. Finally, we present ConceptEdit-Bench, a granular evaluation suite designed to diagnose model capabilities across a vast array of real-world scenarios.

cs.CV↗

CForce: Boosting Parallel Decoding for dLLMs via Consistency Forcing

Diffusion large language models (dLLMs) accelerate language generation by predicting multiple masks in a single forward pass. However, existing dLLMs can suffer from unreliable predictions in early denoising stages under aggressive parallelism strategies, leading to errors that can propagate to later stages. To tackle this issue, we present Consistency Forcing (CForce) for dLLMs, a distillation method to force the mask predictions of early stages to align with those of later stages. CForce trains the model on pre-collected self-rollout trajectories, thereby improving training-inference alignment. We introduce Confidence Adaptive KL Divergence as a distillation objective to conjoin the merits of forward and reverse KL. We further provide a theoretical analysis for the consistency objective to explain why CForce can approximately minimize the prediction error of early stages. Critically, the same formulation applies to both mask-to-token decoding and edit-capable decoding; in the edit-capable case, later token-to-token refinements provide additional supervision for earlier masked-state predictions. Experiments on non-edit and edit-capable LLaDA models show improved speed-quality trade-offs, especially under high-parallelism decoding budgets. Code is available at: https://github.com/inclusionAI/dFactory.

cs.LG↗

LLaDA MoE v2: Scaling Mixture-of-Experts Diffusion Language Models

Diffusion language models (dLLMs) offer an alternative to autoregressive (AR) language modeling, yet the scaling behavior of Mixture-of-Experts (MoE) dLLMs remains poorly understood. We systematically characterize how optimization hyperparameters, compute allocation, and architecture scale for MoE dLLMs, identifying quantitative differences from scaling trends previously reported for AR models. Specifically, for optimization, the optimal nominal batch size grows faster, while the optimal learning rate decays more rapidly with compute. For model--data allocation, IsoFLOP analysis reveals a slight data-side tilt: the optimal token budget grows faster than activated model-side computation. For MoE architecture, larger scales increasingly favor larger expert pools at fixed activated capacity, while moderate expert granularity remains consistently effective and the preferred fraction of activated capacity assigned to shared experts remains stable across scales. Guided by these findings, we train LLaDA MoE v2, a 30B-A3B dLLM, from scratch on 23.5T tokens. With approximately 65\% as many pretraining tokens as Qwen3, LLaDA MoE v2 approaches Qwen3 on several knowledge, reasoning, and coding benchmarks. After supervised fine-tuning alone, it outperforms SDAR Chat on seven of eight reasoning and coding benchmarks and remains close to Qwen3 on several tasks. These results establish practical scaling laws and design principles for MoE dLLMs.

cs.AI↗

Joint Chirp Parameter Selection and Low-Complexity MMSE Receiver Design for AFDM Systems

Affine frequency division multiplexing (AFDM) has emerged as a promising waveform against doubly selective channels under high-mobility communication scenarios. Optimal chirp parameter selection and reduced-complexity receiver design in AFDM are essential for achieving satisfactory bit error rate (BER) performance with low computational complexity. In this paper, we investigate the joint optimization of chirp-parameter selection and low-complexity minimum mean square error (MMSE)-based receiver design by exploiting the structural characteristics of the AFDM effective channel matrix (ECM). First, a simplified BER performance metric is derived by leveraging the diagonal and circulant structure of the discrete affine Fourier transformation (DAFT), based on which a fast circulant-diagonal aggregation (FCDA) algorithm is developed for efficient $c_1$ selection. Then, a low-complexity banded MMSE (LC-BMMSE) receiver is developed by constructing a cyclic-banded ECM through path-wise structured sparsification, where banded Cholesky factorization is employed to avoid direct matrix inversion. Building upon the proposed BER metric and the LC-BMMSE receiver, a hierarchical-search-based joint chirp parameter and structured sparsification (HS-JCPS) algorithm is further proposed to jointly optimize the chirp parameter and sparsification pattern under a given complexity constraint. Simulation results demonstrate that the proposed FCDA reduces the search time for the optimal $c_1$ by an order of magnitude compared with using a BER-based criterion. Moreover, the proposed HS-JCPS algorithm with the LC-BMMSE receiver can identify a near-optimal $c_1$, while attaining a superior performance-complexity tradeoff.

eess.SP↗

AFDM-FTN: A Spectrally Efficient Waveform for High-Mobility Communications

This paper proposes an affine frequency division multiplexing (AFDM)-aided faster-than-Nyquist (FTN) waveform, termed AFDM-FTN, to enhance spectral efficiency (SE) in high-mobility communication scenarios. We first derive the AFDM-FTN input-output relationship and analyze the FTN-induced interference pattern in AFDM-FTN. To address the channel estimation challenges, a low-complexity channel estimator based on the basis expansion model (BEM) is developed. By exploiting the intrinsic characteristics of the AFDM channel matrix and the FTN coefficient matrix, a multi-layer message passing (MLMP) algorithm is proposed that leverages the sparsity of the time-domain (TD) channel and the FTN coefficient matrix, where belief messages are iteratively propagated across the TD channel, FTN, and transform layers. Building upon the BEM-assisted channel estimation and MLMP, a low-complexity joint channel estimation and data detection scheme (BEM-MLMP-JCED) is further developed to iteratively refine channel estimation with the aid of transmitted data. Finally, the channel estimation lower bound, the mean square error (MSE) performance of the BEM-MLMP-JCED, and the computational complexity are analyzed. Simulation results demonstrate that the proposed AFDM-FTN system with BEM-MLMP-JCED achieves comparable BER to conventional AFDM while providing enhanced SE and reduced complexity compared to benchmark receivers.

eess.SP↗

Two-Stage Multi-Modal Fusion with Adaptive Alignment for Action Quality Assessment

Action Quality Assessment (AQA) aims to evaluate how well a person performs a movement, which is essential in applications such as sports scoring, skill assessment, and healthcare. However, unimodal approaches often struggle to capture subtle cues of movement quality in real-world settings. Although multi-modal inputs provide complementary information, existing methods still face two major challenges: heterogeneous modalities often lead to cross-modal misalignment and unstable fusion, and reliable multi-modal annotation is costly, resulting in limited dataset diversity. To address these challenges, we propose DualAlign, a two-stage multi-modal fusion framework with adaptive alignment. The framework first constructs a coherent visual representation by maximizing shared structural information across RGB video, optical flow, and skeleton modalities. Textual semantics are then incorporated after visual stabilization, allowing high-level descriptions to complement rather than distort the underlying visual manifold. To evaluate the framework under realistic multi-modal conditions, we introduce MM--JDM, a movement-quality assessment dataset integrating RGB videos, optical flow, skeleton sequences, and structured text. MM--JDM naturally exhibits modality noise, class imbalance, and label scarcity, making it a challenging benchmark for studying multi-modal fusion and alignment. Extensive experiments show that DualAlign improves average correlation on MM--JDM by 21.16% over the state-of-the-art methods and achieves gains of 3.53% and 5.95% on the RG and Fis-V benchmarks, respectively. DualAlign also remains robust under missing-modality and label-scarce conditions.

cs.CV↗

Chiral three-nucleon forces for the new local position-space two-nucleon potential in $\textit{ab initio}$ many-body calculations

Three-nucleon force (3NF) plays an important role in understanding the structure of finite nuclei and the saturation properties of infinite nuclear matter. More specifically, 3NF should be necessary for each two-nucleon force (2NF) to obtain more accurate description of nuclear systems. 3NF derived from the chiral effective field theory has been successful in $\textit{ab initio}$ calculations of atomic nuclei. Most of established chiral nuclear forces have a nonlocal form in the momentum space. In this work, we construct a companion chiral 3NF specifically tailored to the new Idaho local position-space 2NF, and calculate binding energies and radii of nuclei up to $^{132}$Sn. We find that a chiral 3NF with hybrid local and nonlocal regulators has advantages in improving the nuclear structure calculations of both binding energies and radii with the new Idaho 2NF. The two low-energy constants of 3NF are constrained by the ground-state energies of $^3$H and $^{16}$O as suggested in a recent work.

nucl-th↗