1 citations · 1 across the 17 of their papers we have counts for
17 papers
Semantic DLM+: Improving Diffusion Language Models through Bias-variance Trade-off in Transition Kernel Design
Keyue Jiang, Yuxiang Wang, Yanan Zhao +7
Diffusion Language Models (DLMs) have demonstrated strong scaling capacity as alternatives to autoregressive language models. However, their performance is highly sensitive to the…
Scalable Distributed Learning-Based MPC for Heterogeneous Consumer-Prosumer Building Aggregations: MPC-Aware Feature Selection and Convex Constraint-Coupled Optimization
Kaipeng Xu, Zhuo Zhi, Keyue Jiang
Buildings provide demand-response flexibility through controllable thermal loads and thermal inertia, while distributed photovoltaic deployment creates opportunities to share surpl…
GDSD: Reinforcement Learning as Guided Denoiser Self-Distillation for Diffusion Language Models
Xiaohang Tang, Keyue Jiang, Che Liu +4
Reinforcement learning (RL) can be used to improve the policy (denoiser) of diffusion large language models (dLLMs), while being hindered by the intractability of the policy likeli…
On the Trainability of Masked Diffusion Language Models via Blockwise Locality
Yuxiang Wang, Yu Xiang, Baojian Zhou +4
Masked diffusion language models (MDMs) have recently emerged as a promising alternative to standard autoregressive large language models (AR-LLMs), yet their optimization can be s…
End-to-End Differentiable Predictive Control with Probabilistic Constraint-Satisfaction Guarantees for Building Demand Response
Kaipeng Xu, Zhuo Zhi, Ruixuan Zhao +1
Building Demand Response (DR) benefits from Model Predictive Control (MPC) because thermal flexibility can be optimized over a future horizon while enforcing operational constraint…
Input Convex Encoder-Only Transformer for Computationally Efficient Model Predictive Control in Building Demand Response
Kaipeng Xu, Zhuo Zhi, Keyue Jiang
Learning-based Model Predictive Control (MPC) has emerged as a powerful strategy for building demand response (DR). However, its practical deployment is often hindered by the non-c…
Impact of Connectivity on Laplacian Representations in Reinforcement Learning
Tommaso Giorgi, Pierriccardo Olivieri, Keyue Jiang +2
Learning compact state representations in Markov Decision Processes (MDPs) has proven crucial for addressing the curse of dimensionality in large-scale reinforcement learning (RL)…
Logics-STEM: Empowering LLM Reasoning via Failure-Driven Post-Training and Document Knowledge Enhancement
Mingyu Xu, Cheng Fang, Keyue Jiang +16
We present Logics-STEM, a state-of-the-art reasoning model fine-tuned on Logics-STEM-SFT-Dataset, a high-quality and diverse dataset at 10M scale that represents one of the largest…
LGDC: Latent Graph Diffusion via Spectrum-Preserving Coarsening
Nagham Osman, Keyue Jiang, Davide Buffelli +2
Graph generation is a critical task across scientific domains. Existing methods fall broadly into two categories: autoregressive models, which iteratively expand graphs, and one-sh…
Adaptive Multi-view Graph Contrastive Learning via Fractional-order Neural Diffusion Networks
Yanan Zhao, Feng Ji, Jingyang Dai +4
Graph contrastive learning (GCL) learns node and graph representations by contrasting multiple views of the same graph. Existing methods typically rely on fixed, handcrafted views-…
On the Importance of Task Complexity in Evaluating LLM-Based Multi-Agent Systems
Bohan Tang, Huidong Liang, Keyue Jiang +1
Large language model multi-agent systems (LLM-MAS) offer a promising paradigm for harnessing collective intelligence to achieve more advanced forms of AI behaviour. While recent st…
A Stage-Aware Mixture of Experts Framework for Neurodegenerative Disease Progression Modelling
Tiantian He, Keyue Jiang, An Zhao +5
The long-term progression of neurodegenerative diseases is commonly conceptualized as a spatiotemporal diffusion process that consists of a graph diffusion process across the struc…
Bures-Wasserstein Flow Matching for Graph Generation
Keyue Jiang, Jiahao Cui, Xiaowen Dong +1
Graph generation has emerged as a critical task in fields ranging from drug discovery to circuit design. Contemporary approaches, notably diffusion and flow-based models, have achi…
Heterogeneous Graph Structure Learning through the Lens of Data-generating Processes
Keyue Jiang, Bohan Tang, Xiaowen Dong +1
Inferring the graph structure from observed data is a key task in graph machine learning to capture the intrinsic relationship between data entities. While significant advancements…
Effects of Dropout on Performance in Long-range Graph Learning Tasks
Jasraj Singh, Keyue Jiang, Brooks Paige +1
Message Passing Neural Networks (MPNNs) are a class of Graph Neural Networks (GNNs) that propagate information across the graph via local neighborhoods. The scheme gives rise to tw…
Training-Free Message Passing for Learning on Hypergraphs
Bohan Tang, Zexi Liu, Keyue Jiang +2
Hypergraphs are crucial for modelling higher-order interactions in real-world data. Hypergraph neural networks (HNNs) effectively utilise these structures by message passing to gen…
A Markov Random Field model for Hypergraph-based Machine Learning
Bohan Tang, Keyue Jiang, Laura Toni +2
Understanding the data-generating process is essential for building machine learning models that generalise well while ensuring robustness and interpretability. This paper addresse…