← Search

Lei Bai

101 accepted papers

2026

Are Tools Always Beneficial? Learning to Invoke Tools Adaptively for Dual-Mode Multimodal LLM Reasoning

ICML 2026poster

Tool-augmented reasoning has emerged as a promising direction for enhancing the reasoning capabilities of multimodal large language models (MLLMs). However, existing studies mainly focus on enabling models to perform tool invocation, while neglecting the necessity of invoking tools. We argue that to…

Cited by 0SourceScholar
2026

Benchmarking Physics-Informed Time-Series Models for Operational Global Station Weather Forecasting

ICML 2026poster

The development of Time-Series Forecasting (TSF) models is often constrained by the lack of comprehensive datasets, especially in Global Station Weather Forecasting (GSWF), where existing datasets are small, temporally short, and spatially sparse. To address this, we introduce WEATHER-5K, a large-sc…

Cited by 0SourceScholar
2026

Beyond Gemini-3-Pro: Revisiting LLM Routing and Aggregation at Scale

ICML 2026poster

Large Language Models (LLMs) have rapidly advanced, with Gemini-3-Pro setting a new performance milestone. In this work, we explore collective intelligence as an alternative to monolithic scaling, and demonstrate that open-source LLMs' collaboration can surpass Gemini-3-Pro. We first revisit LLM rou…

Cited by 0SourceScholar
2026

CMPhysBench: A Benchmark for Evaluating Large Language Models in Condensed Matter Physics

ICLR 2026poster

We introduce CMPhysBench, designed to assess the proficiency of Large Language Models (LLMs) in Condensed Matter Physics, as a novel Benchmark. CMPhysBench is composed of more than 520 graduate-level meticulously curated questions covering both representative subfields and foundational theoretical f…

Cited by 0SourcecodeScholar
2026

CoMAS: Co-Evolving Multi-Agent Systems via Interaction Rewards

ICLR 2026poster

Self-evolution is a central research topic in enabling large language model (LLM)-based agents to continually improve their capabilities after pretraining. Recent research has witnessed a transition from reinforcement learning (RL)-free to RL-based methods. Current RL-based methods either rely on de…

Cited by 0SourcecodeScholar
2026

CompBench: Benchmarking Complex Instruction-guided Image Editing

CVPR 2026

While real-world applications increasingly demand intricate scene manipulation, existing instruction-guided image editing benchmarks often oversimplify task complexity and lack comprehensive, fine-grained instructions. To bridge this gap, we introduce CompBench, a large-scale benchmark specifically

Cited by 0SourcecodeScholar
2026

EMFormer: Efficient Multi-Scale Transformer for Accumulative Context Weather Forecasting

ICML 2026poster

Long-term weather forecasting is critical for socioeconomic planning and disaster preparedness. While recent approaches employ finetuning to extend prediction horizons, they remain constrained by the issues of catastrophic forgetting, error accumulation, and high training overhead. To address these …

Cited by 0SourceScholar
2026

Earth-Agent: Unlocking the Full Landscape of Earth Observation with Agents

ICLR 2026poster

Earth observation (EO) is essential for understanding the evolving states of the Earth system. Although recent MLLMs have advanced EO research, they still lack the capability to tackle complex tasks that require multi-step reasoning and the use of domain-specific tools. Agent-based methods offer a…

Cited by 0SourcecodeScholar
2026

EarthSE: A Benchmark Evaluating Earth Scientific Exploration Capability for Large Language Models

ICLR 2026poster

Advancements in Large Language Models (LLMs) drive interest in scientific applications, necessitating specialized benchmarks such as Earth science. Existing benchmarks either present a general science focus devoid of Earth science specificity or cover isolated subdomains, lacking holistic evaluation…

Cited by 0SourceScholar
2026

Eigen-1: Scientific Reasoning through Adaptive Multi-Agent Refinement and Monitor-based RAG

ICLR 2026poster

Large language models (LLMs) have recently shown strong progress on scientific reasoning, yet two major bottlenecks remain. First, explicit retrieval fragments reasoning, imposing a hidden tool tax of extra tokens and steps. Second, multi-agent pipelines often dilute strong solutions by averaging ac…

Cited by 0SourcecodeScholar
2026

FinPercep-RM: A Fine-grained Reward Model and Co-evolutionary Curriculum for RL-based Real-world Super-Resolution

CVPR 2026

Inspired by the success of Reinforcement Learning with Human Feedback (RLHF) in image generation, recent work has adapted reward-based learning to image super-resolution (ISR) by using Image Quality Assessment (IQA) models as rewards. However, existing IQA models typically output only a single globa

Cited by 0SourcecodeScholar
2026

HiPhO: How Far Are (M)LLMs from Humans in the Latest High School Physics Olympiad Benchmark?

ICML 2026poster

Recently, the physics reasoning capabilities of (M)LLMs have attracted growing attention. However, existing physics benchmarks suffer from two major gaps: they neither provide systematic and up-to-date coverage of physics Olympiads, nor enable direct performance comparison with humans. To bridge the…

Cited by 0SourceScholar
2026

LabBuilder: Protocol-Grounded 3D Layout Generation for Interactable and Safe Laboratory

ICML 2026poster

Automated laboratories hold the promise of accelerating scientific discovery, yet their deployment is bottlenecked by the difficulty of designing safe and executable environments. While simulator-based design offers scalability, existing 3D scene generation methods are primarily tailored for househo…

Cited by 0SourceScholar
2026

ML-Agent: Reinforcing LLM Agents for Autonomous Machine Learning Engineering

ICML 2026poster

The emergence of large language model (LLM)-based agents has significantly advanced the development of autonomous machine learning (ML) engineering. However, the dominant prompt-based paradigm exhibits limitations: smaller models lack the capacity to learn from execution trajectories for generalizat…

Cited by 0SourcecodeScholar
2026

MME-Reasoning: A Broad-Spectrum Benchmark for Evaluating Logical Reasoning in MLLMs

ICML 2026poster

Logical reasoning is a fundamental aspect of human intelligence and an essential capability for multimodal large language models (MLLMs). Despite the significant advancement in multimodal reasoning, existing benchmarks fail to comprehensively evaluate their reasoning abilities due to the lack of exp…

Cited by 0SourceScholar
2026

Omni-Weather: Unified Multimodal Foundation Model for Weather Generation and Understanding

ICLR 2026poster

Weather modeling requires both accurate prediction and mechanistic interpretation, yet existing methods treat these goals in isolation, separating generation from understanding. To address this gap, we present Omni-Weather, the first multimodal foundation model that unifies weather generation and un…

Cited by 0SourcecodeScholar
2026

Order Matters in Retrosynthesis: Structure-aware Generation via Reaction-Center-Guided Discrete Flow Matching

ICML 2026poster

Template-free retrosynthesis methods treat the task as black-box sequence generation, limiting learning efficiency, while semi-template approaches rely on rigid reaction libraries that constrain generalization. We address this gap with a key insight: atom ordering in neural representations matters. …

Cited by 0SourceScholar
2026

SCI-Verifier: Scientific Verifier with Thinking

ICLR 2026poster

As large language models (LLMs) are increasingly applied to scientific reasoning, the complexity of answer formats and the diversity of equivalent expressions make answer verification a critical yet challenging task. Existing verification studies in scientific domains suffer from two major limitatio…

Cited by 0SourcecodeScholar
2026

STCast: Adaptive Boundary Alignment for Global and Regional Weather Forecasting

CVPR 2026

To gain finer regional forecasts, many works have explored the regional integration from the global atmosphere, e.g., by solving boundary equations in physics-based methods or cropping regions from global forecasts in data-driven methods. However, the effectiveness of these methods is often constrai

Cited by 0SourcecodeScholar
2026

SciAgentGym: Benchmarking Multi-Step Scientific Tool-Use in LLM Agents

ICML 2026poster

Scientific reasoning inherently demands integrating sophisticated toolkits to navigate domain-specific knowledge. Yet, current benchmarks largely overlook agents' ability to orchestrate tools for such rigorous workflows. To bridge this gap, we introduce **SciAgentGym**, a scalable interactive enviro…

Cited by 0SourceScholar
2026

Squeeze the Soaked Sponge: Efficient Off-policy RFT for Large Language Model

ICLR 2026poster

Reinforcement Learning (RL) has demonstrated its potential to improve the reasoning ability of Large Language Models (LLMs), yet most existing Reinforcement Finetuning (RFT) methods are inherently \textit{on-policy} RL, failing to reuse historical data and thus preventing efficient scaling. In this…

Cited by 0SourceScholar
2026

SynWeather: Weather Observation Data Synthesis Across Multiple Regions and Variables via a General Diffusion Transformer

AAAI 2026technical

With the advancement of meteorological instruments, abundant data has become available. However, due to instruments’ intrinsic limitations such as environmental sensitivity and orbital constraints, raw data often suffer from temporal or spatial gaps, making it urgent to leverage data synthesis tech

Cited by 0SourcePDFScholar
2026

The Avengers: A Routing Recipe for Collective Intelligence in Language Models

AAAI 2026technical

Proprietary models are increasingly dominating the race for ever-larger language models. Can open-source, smaller models remain competitive across a broad range of tasks? In this paper, we present the Avengers---a lightweight framework that leverages the collective intelligence of these smaller mod

Cited by 0SourcePDFScholar
2026

Transforming Weather Data from Pixel to Latent Space

ICML 2026oral

The increasing impact of climate change and extreme weather events has spurred growing interest in deep learning for weather research. However, existing studies often rely on weather data in pixel space, which presents several challenges such as smooth outputs in model outputs, limited applicability…

Cited by 0SourceScholar
2026

Transition Models: Rethinking the Generative Learning Objective

CVPR 2026

A fundamental dilemma in generative modeling persists: iterative diffusion models achieve outstanding fidelity, but at a significant computational cost, while efficient few-step alternatives are constrained by a hard quality ceiling. This conflict between generation steps and output quality arises f

Cited by 0SourcecodeScholar
2026

Unleashing Scientific Reasoning for Bio-experimental Protocol Generation via Structured Component-based Reward Mechanism

ICLR 2026poster

The foundation of reproducible science lies in protocols that are precise, logically ordered, and executable. The autonomous generation of these protocols through natural language queries could greatly improve the efficiency of the reproduction process. However, current leading large language models…

Cited by 0SourcecodeScholar
2025

Actial: Activate Spatial Reasoning Ability of Multimodal Large Language Models

NeurIPS 2025poster

Recent advances in Multimodal Large Language Models (MLLMs) have significantly improved 2D visual understanding, prompting interest in their application to complex 3D reasoning tasks. However, it remains unclear whether these models can effectively capture the detailed spatial information required f…

Cited by 0SourceScholar
2025

Align-DA: Align Score-based Atmospheric Data Assimilation with Multiple Preferences

NeurIPS 2025poster

Data assimilation (DA) aims to estimate the full state of a dynamical system by combining partial and noisy observations with a prior model forecast, commonly referred to as the background. In atmospheric applications, this problem is fundamentally ill-posed due to the sparsity of observations relat…

Cited by 0SourceScholar
2025

BMMR: A Large-Scale Bilingual Multimodal Multi-Discipline Reasoning Dataset

NeurIPS 2025poster

In this paper, we introduce BMMR, a large-scale bilingual, multimodal, multi-disciplinary reasoning dataset for the community to develop and evaluate large multimodal models (LMMs). BMMR comprises 100k university-level questions drawn from 300 UNESCO-defined subjects, spanning diverse formats—multip…

Cited by 0SourceScholar
2025

Breaking the Compression Ceiling: Data-Free Pipeline for Ultra-Efficient Delta Compression

NeurIPS 2025poster

With the rise of the fine-tuned–pretrained paradigm, storing numerous fine-tuned models for multi-tasking creates significant storage overhead. Delta compression alleviates this by storing only the pretrained model and the highly compressed delta weights (the differences between fine-tuned and pretr…

Cited by 0SourcecodeScholar
2025

Chimera: Improving Generalist Model with Domain-Specific Experts

ICCV 2025poster

Large Multi-modal Models (LMMs), trained on web-scale datasets predominantly composed of natural images, have demonstrated remarkable performance on general tasks. However, these models often exhibit limited specialized capabilities for domain-specific tasks that require extensive domain prior knowl…

Cited by 0SourcePDFScholar
2025

ComfyBench: Benchmarking LLM-based Agents in ComfyUI for Autonomously Designing Collaborative AI Systems

CVPR 2025poster

Much previous AI research has focused on developing monolithic models to maximize their intelligence, with the primary goal of enhancing performance on specific tasks. In contrast, this work attempts to study using LLM-based agents to design collaborative AI systems autonomously. To explore this pro…

2025

DAWP: A framework for global observation forecasting via Data Assimilation and Weather Prediction in satellite observation space

NeurIPS 2025poster

Weather prediction is a critical task for human society, where impressive progress has been made by training artificial intelligence weather prediction (AIWP) methods with reanalysis data. However, reliance on reanalysis data limits the AIWPs with shortcomings, including data assimilation biases an…

Cited by 0SourceScholar
2025

DiffSR: Learning Radar Reflectivity Synthesis via Diffusion Model from Satellite Observations

ICASSP 2025accepted

Weather radar data synthesis can fill in data for areas where ground observations are missing. Existing methods often employ reconstruction-based approaches with MSE loss to reconstruct radar data from satellite observation. However, such methods lead to over-smoothing, which hinders the generation…

Cited by 0SourceScholar
2025

Dolphin: Moving Towards Closed-loop Auto-research through Thinking, Practice, and Feedback

ACL 2025long

The scientific research paradigm is undergoing a profound transformation owing to the development of Artificial Intelligence (AI). Recent works demonstrate that various AI-assisted research methods can largely improve research efficiency by improving data analysis, accelerating computation, and fost…

2025

Global Tropical Cyclone Intensity Forecasting with Multi-modal Multi-scale Causal Autoregressive Model

ICASSP 2025accepted

Accurate forecasting of tropical cyclone (TC) intensity is crucial for formulating disaster risk reduction strategies. Current methods predominantly rely on limited spatiotemporal information from ERA5 data and neglect the causal relationships between these physical variables, failing to fully captu…

Cited by 0SourceScholar
2025

IceDiff: High Resolution and High-Quality Arctic Sea Ice Forecasting with Generative Diffusion Prior

CVPR 2025poster

Variation of Arctic sea ice has significant impacts on polar ecosystems, transporting routes, coastal communities, and global climate. Tracing the change of sea ice at a finer scale is paramount for both operational applications and scientific studies. Recent pan-Arctic sea ice forecasting methods t…

2025

InfGen: A Resolution-Agnostic Paradigm for Scalable Image Synthesis

ICCV 2025poster

Arbitrary resolution image generation provides a consistent visual experience across devices, having extensive applications for producers and consumers. Current diffusion models increase computational demand quadratically with resolution, causing 4K image generation delays over 100 seconds. To solve…

2025

LabUtopia: High-Fidelity Simulation and Hierarchical Benchmark for Scientific Embodied Agents

NeurIPS 2025poster

Scientific embodied agents play a crucial role in modern laboratories by automating complex experimental workflows. Compared to typical household environments, laboratory settings impose significantly higher demands on perception of physical-chemical transformations and long-horizon planning, making…

Cited by 0SourcecodeScholar
2025

Latent Harmony: Synergistic Unified UHD Image Restoration via Latent Space Regularization and Controllable Refinement

NeurIPS 2025poster

Ultra-High Definition (UHD) image restoration struggles to balance computational efficiency and detail retention. While Variational Autoencoders (VAEs) offer improved efficiency by operating in the latent space, with the Gaussian variational constraint, this compression preserves semantics but sacri…

Cited by 0SourceScholar
2025

Learning Urban Climate Dynamics via Physics-Guided Urban Surface–Atmosphere Interactions

NeurIPS 2025poster

Urban warming differs markedly from regional background trends, highlighting the unique behavior of urban climates and the challenges they present. Accurately predicting local urban climate necessitates modeling the interactions between urban surfaces and atmospheric forcing. Although off-the-shelf…

Cited by 0SourceScholar
2025

LoRA-EnVar: Parameter-Efficient Hybrid Ensemble Variational Assimilation for Weather Forecasting

NeurIPS 2025poster

Accurate estimation of background error (i.e., forecast error) distribution is critical for effective data assimilation (DA) in numerical weather prediction (NWP). In state-of-the-art operational DA systems, it is common to account for the temporal evolution of background errors by employing hybrid…

Cited by 0SourceScholar
2025

Multi-agent Architecture Search via Agentic Supernet

ICML 2025oral

Large Language Model (LLM)-empowered multi-agent systems extend the cognitive boundaries of individual agents through disciplined collaboration and interaction, while constructing these systems often requires labor-intensive manual designs. Despite the availability of methods to automate the design…

2025

NUTS: Eddy-Robust Reconstruction of Surface Ocean Nutrients via Two-Scale Modeling

NeurIPS 2025poster

Reconstructing ocean surface nutrients from sparse observations is critical for understanding long-term biogeochemical cycles. Most prior work focuses on reconstructing atmospheric fields and treats the reconstruction problem as image inpainting, assuming smooth, single-scale dynamics. In contrast,…

Cited by 0SourceScholar
2025

PostCast: Generalizable Postprocessing for Precipitation Nowcasting via Unsupervised Blurriness Modeling

ICLR 2025poster

Precipitation nowcasting plays a pivotal role in socioeconomic sectors, especially in severe convective weather warnings. Although notable progress has been achieved by approaches mining the spatiotemporal correlations with deep learning, these methods still suffer severe blurriness as the lead time…

Cited by 3SourcePDFScholar
2025

RadarQA: Multi-modal Quality Analysis of Weather Radar Forecasts

NeurIPS 2025poster

Quality analysis of weather forecasts is an essential topic in meteorology. Although traditional score-based evaluation metrics can quantify certain forecast errors, they are still far from meteorological experts in terms of descriptive capability, interpretability, and understanding of dynamic evol…

Cited by 0SourceScholar
2025

ReSo: A Reward-driven Self-organizing LLM-based Multi-Agent System for Reasoning Tasks

EMNLP 2025

Multi-agent systems have emerged as a promising approach for enhancing the reasoning capabilities of large language models in complex problem-solving. However, current MAS frameworks are limited by poor flexibility and scalability, with underdeveloped optimization strategies. To address these challe

2025

Reinventing Multi-Agent Collaboration through Gaussian-Image Synergy in Diffusion Policies

NeurIPS 2025poster

Despite significant advances in robotic policy generation, effective coordination in embodied multi-agent systems remains a fundamental challenge—particularly in scenarios where agents must balance individual perspectives with global environmental awareness. Existing approaches often struggle to bal…

Cited by 0SourceScholar
2025

Retro-R1: LLM-based Agentic Retrosynthesis

NeurIPS 2025poster

Retrosynthetic planning is a fundamental task in chemical discovery. Due to the vast combinatorial search space, identifying viable synthetic routes remains a significant challenge--even for expert chemists. Recent advances in Large Language Models (LLMs), particularly equipped with reinforcement le…

Cited by 0SourceScholar
2025

SIFusion: A Unified Fusion Framework for Multi-granularity Arctic Sea Ice Forecasting

NeurIPS 2025poster

Arctic sea ice performs a vital role in global climate and has paramount impacts on both polar ecosystems and coastal communities. In the last few years, multiple deep learning based pan-Arctic sea ice concentration (SIC) forecasting methods have emerged and showcased superior performance over physi…

Cited by 0SourceScholar
2025

SURVEYFORGE : On the Outline Heuristics, Memory-Driven Generation, and Multi-dimensional Evaluation for Automated Survey Writing

ACL 2025long

Survey paper plays a crucial role in scientific research, especially given the rapid growth of research publications. Recently, researchers have begun using LLMs to automate survey generation for better efficiency. However, the quality gap between LLM-generated surveys and those written by human rem…

2025

Satellite Observations Guided Diffusion Model for Accurate Meteorological States at Arbitrary Resolution

CVPR 2025highlight

Accurate acquisition of surface meteorological conditions at arbitrary locations holds significant importance for weather forecasting and climate simulation. Meteorological states derived from satellite observations are often provided in the form of low-resolution grid fields. If spatial interpolati…

2025

Scaling Physical Reasoning with the PHYSICS Dataset

NeurIPS 2025poster

Large Language Models (LLMs) have achieved remarkable progress on advanced reasoning tasks such as mathematics and coding competitions. Meanwhile, physics, despite being both reasoning-intensive and essential to real-world understanding, received limited academic and industrial attention. This paper…

Cited by 0SourcecodeScholar
2025

Scientists' First Exam: Probing Cognitive Abilities of MLLM via Perception, Understanding, and Reasoning

NeurIPS 2025poster

Scientific discoveries increasingly rely on complex multimodal reasoning based on information-intensive scientific data and domain-specific expertise. Empowered by expert-level scientific benchmarks, scientific Multimodal Large Language Models (MLLMs) hold the potential to significantly enhance this…

Cited by 0SourceScholar
2025

TimeKAN: KAN-based Frequency Decomposition Learning Architecture for Long-term Time Series Forecasting

ICLR 2025poster

Real-world time series often have multiple frequency components that are intertwined with each other, making accurate time series forecasting challenging. Decomposing the mixed frequency components into multiple single frequency components is a natural choice. However, the information density of pat…

2025

Understand Before You Generate: Self-Guided Training for Autoregressive Image Generation

NeurIPS 2025poster

Recent studies have demonstrated the importance of high-quality visual representations in image generation and have highlighted the limitations of generative models in image understanding. As a generative paradigm originally designed for natural language, autoregressive models face similar challenge…

Cited by 0SourceScholar
2025

UniSTD: Towards Unified Spatio-Temporal Learning across Diverse Disciplines

CVPR 2025poster

Traditional spatiotemporal models generally rely on task-specific architectures, which limit their generalizability and scalability across diverse tasks due to domain-specific design requirements. In this paper, we introduce UniSTD, a unified Transformer-based framework for spatiotemporal modeling,…

2025

VA-MoE: Variables-Adaptive Mixture of Experts for Incremental Weather Forecasting

ICCV 2025poster

This paper presents Variables-Adaptive Mixture of Experts (VA-MoE), a novel framework for incremental weather forecasting that dynamically adapts to evolving spatiotemporal patterns in real-time data. Traditional weather prediction models often struggle with exorbitant computational expenditure and…

2025

VAE-Var: Variational Autoencoder-Enhanced Variational Methods for Data Assimilation in Meteorology

ICLR 2025poster

Data assimilation (DA) is an essential statistical technique for generating accurate estimates of a physical system's states by combining prior model predictions with observational data, especially in the realm of weather forecasting. Effectively modeling the prior distribution while adapting to div…

Cited by 1SourcePDFScholar
2025

VIKI‑R: Coordinating Embodied Multi-Agent Cooperation via Reinforcement Learning

NeurIPS 2025poster

Coordinating multiple embodied agents in dynamic environments remains a core challenge in artificial intelligence, requiring both perception-driven reasoning and scalable cooperation strategies. While recent works have leveraged large language models (LLMs) for multi-agent planning, a few have begun…

Cited by 0SourceScholar
2025

VLIPP: Towards Physically Plausible Video Generation with Vision and Language Informed Physical Prior

ICCV 2025accepted

Video diffusion models (VDMs) have advanced significantly in recent years, enabling the generation of highly realistic videos and drawing the attention of the community in their potential as world simulators. However, despite their capabilities, VDMs often fail to produce physically plausible videos…

2025

VQLTI: Long-Term Tropical Cyclone Intensity Forecasting with Physical Constraints

AAAI 2025technical

Tropical cyclone (TC) intensity forecasting is crucial for early disaster warning and emergency decision-making. Numerous researchers have explored deep-learning methods to address computational and post-processing issues in operational forecasting. Regrettably, they exhibit subpar long-term forecas…

2025

WeatherGFM: Learning a Weather Generalist Foundation Model via In-context Learning

ICLR 2025poster

The Earth's weather system involves intricate weather data modalities and diverse weather understanding tasks, which hold significant value to human life. Existing data-driven models focus on single weather understanding tasks (e.g., weather forecasting). While these models have achieved promising…

2025

WorldSimBench: Towards Video Generation Models as World Simulators

ICML 2025poster

Recent advancements in predictive models have demonstrated exceptional capabilities in predicting the future state of objects and scenes. However, the lack of categorization based on inherent characteristics continues to hinder the progress of predictive model development. Additionally, existing ben…

Cited by 18SourcePDFScholar
2025

scMRDR: A scalable and flexible framework for unpaired single-cell multi-omics data integration

NeurIPS 2025spotlight

Advances in single-cell sequencing have enabled high-resolution profiling of diverse molecular modalities, while integrating unpaired multi-omics single-cell data remains challenging. Existing approaches either rely on pair information or prior correspondences, or require computing a global pairwise…

Cited by 0SourceScholar
2024

Autoregressive Omni-Aware Outpainting for Open-Vocabulary 360-Degree Image Generation

AAAI 2024technical

A 360-degree (omni-directional) image provides an all-encompassing spherical view of a scene. Recently, there has been an increasing interest in synthesising 360-degree images from conventional narrow field of view (NFoV) images captured by digital cameras and smartphones, for providing immersive ex…

2024

CasCast: Skillful High-resolution Precipitation Nowcasting via Cascaded Modelling

ICML 2024poster

Precipitation nowcasting based on radar data plays a crucial role in extreme weather prediction and has broad implications for disaster management. Despite progresses have been made based on deep learning, two key challenges of precipitation nowcasting are not well-solved: (i) the modeling of comple…

Cited by 20SourcePDFScholar
2024

FNP: Fourier Neural Processes for Arbitrary-Resolution Data Assimilation

NeurIPS 2024poster

Data assimilation is a vital component in modern global medium-range weather forecasting systems to obtain the best estimation of the atmospheric state by combining the short-term forecast and observations. Recently, AI-based data assimilation approaches have attracted increasing attention for their…

2024

FiT: Flexible Vision Transformer for Diffusion Model

ICML 2024spotlight

In the context of this reality, existing diffusion models, such as Diffusion Transformers, often face challenges when processing image resolutions outside of their trained domain. To overcome this limitation, we present the Flexible Vision Transformer (FiT), a transformer architecture specifically d…

2024

Generalizing Weather Forecast to Fine-grained Temporal Scales via Physics-AI Hybrid Modeling

NeurIPS 2024poster

Data-driven artificial intelligence (AI) models have made significant advancements in weather forecasting, particularly in medium-range and nowcasting. However, most data-driven weather forecasting models are black-box systems that focus on learning data mapping rather than fine-grained physical evo…

2024

Instruct-ReID: A Multi-purpose Person Re-identification Task with Instructions

CVPR 2024poster

Human intelligence can retrieve any person according to both visual and language descriptions. However the current computer vision community studies specific person re-identification (ReID) tasks in different scenarios separately which limits the applications in the real world. This paper strives to…

2024

MotionGPT: Finetuned LLMs Are General-Purpose Motion Generators

AAAI 2024technical

Generating realistic human motion from given action descriptions has experienced significant advancements because of the emerging requirement of digital humans. While recent works have achieved impressive results in generating motion directly from textual action descriptions, they often support only…

2024

Non-Neighbors Also Matter to Kriging: A New Contrastive-Prototypical Learning

AISTATS 2024poster

Kriging aims to estimate the attributes of unseen geo-locations from observations in the spatial vicinity or physical connections. Existing works assume that neighbors’ information offers the basis for estimating the unobserved target while ignoring non-neighbors. However, neighbors could also be qu…

2024

On Learning Multi-Modal Forgery Representation for Diffusion Generated Video Detection

NeurIPS 2024poster

Large numbers of synthesized videos from diffusion models pose threats to information security and authenticity, leading to an increasing demand for generated content detection. However, existing video-level detection algorithms primarily focus on detecting facial forgeries and often fail to identif…

2024

Towards Dynamic Spatial-Temporal Graph Learning: A Decoupled Perspective

AAAI 2024technical

With the progress of urban transportation systems, a significant amount of high-quality traffic data is continuously collected through streaming manners, which has propelled the prosperity of the field of spatial-temporal graph prediction. In this paper, rather than solely focusing on designing pow…

Cited by 20SourcePDFScholar
2024

Towards a Self-contained Data-driven Global Weather Forecasting Framework

ICML 2024poster

Data-driven weather forecasting models are advancing rapidly, yet they rely on initial states (i.e., analysis states) typically produced by traditional data assimilation algorithms. Four-dimensional variational assimilation (4DVar) is one of the most widely adopted data assimilation algorithms in nu…

Cited by 8SourcePDFScholar
2023

Cycle-consistent Masked AutoEncoder for Unsupervised Domain Generalization

ICLR 2023poster

Self-supervised learning methods undergo undesirable performance drops when there exists a significant domain gap between training and testing scenarios. Therefore, unsupervised domain generalization (UDG) is proposed to tackle the problem, which requires the model to be trained on several different…

Cited by 7SourcePDFScholar
2023

FEND: A Future Enhanced Distribution-Aware Contrastive Learning Framework for Long-Tail Trajectory Prediction

CVPR 2023poster

Predicting the future trajectories of the traffic agents is a gordian technique in autonomous driving. However, trajectory prediction suffers from data imbalance in the prevalent datasets, and the tailed data is often more complicated and safety-critical. In this paper, we focus on dealing with the…

2023

GReTo: Remedying dynamic graph topology-task discordance via target homophily

ICLR 2023poster

Dynamic graphs are ubiquitous across disciplines where observations usually change over time. Regressions on dynamic graphs often contribute to diverse critical tasks, such as climate early-warning and traffic controlling. Existing homophily Graph Neural Networks (GNNs) adopt physical connections or…

Cited by 36SourcePDFScholar
2023

HumanBench: Towards General Human-Centric Perception With Projector Assisted Pretraining

CVPR 2023poster

Human-centric perceptions include a variety of vision tasks, which have widespread industrial applications, including surveillance, autonomous driving, and the metaverse. It is desirable to have a general pretrain model for versatile human-centric downstream tasks. This paper forges ahead along this…

2023

LAMM: Language-Assisted Multi-Modal Instruction-Tuning Dataset, Framework, and Benchmark

NeurIPS 2023poster

Large language models have emerged as a promising approach towards achieving general-purpose AI agents. The thriving open-source LLM community has greatly accelerated the development of agents that support human-machine dialogue interaction through natural language processing. However, human interac…

2023

LargeST: A Benchmark Dataset for Large-Scale Traffic Forecasting

NeurIPS 2023poster

Road traffic forecasting plays a critical role in smart city initiatives and has experienced significant advancements thanks to the power of deep learning in capturing non-linear patterns of traffic data. However, the promising results achieved on current public datasets may not be applicable to pra…

2023

Multi-Scale Control Signal-Aware Transformer for Motion Synthesis without Phase

AAAI 2023technical

Synthesizing controllable motion for a character using deep learning has been a promising approach due to its potential to learn a compact model without laborious feature engineering. To produce dynamic motion from weak control signals such as desired paths, existing methods often require auxiliary…

Cited by 10SourcePDFScholar
2023

STEERER: Resolving Scale Variations for Counting and Localization via Selective Inheritance Learning

ICCV 2023poster

Scale variation is a deep-rooted problem in object counting, which has not been effectively addressed by existing scale-aware algorithms. An important factor is that they typically involve cooperative learning across multi-resolutions, which could be suboptimal for learning the most discriminative f…

Cited by 51PDFcodeScholar
2023

Seeing is not always believing: Benchmarking Human and Model Perception of AI-Generated Images

NeurIPS 2023poster

Photos serve as a way for humans to record what they experience in their daily lives, and they are often regarded as trustworthy sources of information. However, there is a growing concern that the advancement of artificial intelligence (AI) technology may produce fake photos, which can create confu…

2023

Trust Your Partner's Friends: Hierarchical Cross-Modal Contrastive Pre-Training for Video-Text Retrieval

ICASSP 2023accepted

Video-text retrieval has greatly benefited from the massive web video in recent years, while the performance is still limited to the weak supervision from the uncurated data. In this work, we propose to leverage the well-represented information of each original modality and exploit complementary inf…

Cited by 0SourceScholar
2023

UniHCP: A Unified Model for Human-Centric Perceptions

CVPR 2023poster

Human-centric perceptions (e.g., pose estimation, human parsing, pedestrian detection, person re-identification, etc.) play a key role in industrial applications of visual models. While specific human-centric tasks have their own relevant semantic aspect to focus on, they also share the same underly…

2022

Backbone Is All Your Need: A Simplified Architecture for Visual Object Tracking

ECCV 2022poster

"Exploiting a general-purpose neural architecture to replace hand-wired designs or inductive biases has recently drawn extensive interest. However, existing tracking approaches rely on customized sub-modules and need prior knowledge for architecture selection, hindering the development of tracking i…

2022

DR.VIC: Decomposition and Reasoning for Video Individual Counting

CVPR 2022poster

Pedestrian counting is a fundamental tool for understanding pedestrian patterns and crowd flow analysis. Existing works (e.g., image-level pedestrian counting, crossline crowd counting et al.) either only focus on the image-level counting or are constrained to the manual annotation of lines. In this…

Cited by 28PDFcodeScholar
2022

Domain Invariant Masked Autoencoders for Self-Supervised Learning from Multi-Domains

ECCV 2022poster

"Generalizing learned representations across significantly different visual domains is a fundamental yet crucial ability of the human visual system. While recent self-supervised learning methods have achieved good performances with evaluation set on the same domain as the training set, they will hav…

Cited by 19SourcePDFScholar
2022

Fast-MoCo: Boost Momentum-Based Contrastive Learning with Combinatorial Patches

ECCV 2022poster

"Contrastive-based self-supervised learning methods achieved great success in recent years. However, self-supervision requires extremely long training epochs (e.g., 800 epochs for MoCo v3) to achieve promising results, which is unacceptable for the general academic community and hinders the developm…

2022

Relative Contrastive Loss for Unsupervised Representation Learning

ECCV 2022poster

"Defining positive and negative samples is critical for learning visual variations of the semantic classes in an unsupervised manner. Previous methods either construct positive sample pairs as different data augmentations on the same image (i.e., single-instance-positive) or estimate a class prototy…

Cited by 3SourcePDFScholar
2022

Revisiting the Transferability of Supervised Pretraining: An MLP Perspective

CVPR 2022poster

The pretrain-finetune paradigm is a classical pipeline in visual learning. Recent progress on unsupervised pretraining methods shows superior transfer performance to their supervised counterparts. This paper revisits this phenomenon and sheds new light on understanding the transferability gap betwee…

Cited by 72PDFScholar
2022

SepFusion: Finding Optimal Fusion Structures for Visual Sound Separation

AAAI 2022technical

Multiple modalities can provide rich semantic information; and exploiting such information will normally lead to better performance compared with the single-modality counterpart. However, it is not easy to devise an effective cross-modal fusion structure due to the variations of feature dimensions…

Cited by 15SourcePDFScholar
2022

Unifying Visual Contrastive Learning for Object Recognition from a Graph Perspective

ECCV 2022poster

"Recent contrastive based unsupervised object recognition methods leverage a Siamese architecture, which has two branches composed of a backbone, a projector layer, and an optional predictor layer in each branch. To learn the parameters of the backbone, existing methods have a similar projector laye…

Cited by 8SourcePDFScholar
2022

Unsupervised Object Detection Pretraining with Joint Object Priors Generation and Detector Learning

NeurIPS 2022accept

Unsupervised pretraining methods for object detection aim to learn object discrimination and localization ability from large amounts of images. Typically, recent works design pretext tasks that supervise the detector to predict the defined object priors. They normally leverage heuristic methods to p…

Cited by 5SourcePDFScholar
2021

GLiT: Neural Architecture Search for Global and Local Image Transformer

ICCV 2021poster

We introduce the first Neural Architecture Search (NAS) method to find a better transformer architecture for image recognition. Recently, transformers without CNN-based backbones are found to achieve impressive performance for image recognition. However, the transformer is designed for NLP tasks and…

Cited by 131PDFcodeScholar
2021

Graph-Based 3D Multi-Person Pose Estimation Using Multi-View Images

ICCV 2021poster

This paper studies the task of estimating the 3D human poses of multiple persons from multiple calibrated camera views. Following the top-down paradigm, we decompose the task into two stages, i.e. person localization and pose estimation. Both stages are processed in coarse-to-fine manners. And we pr…

Cited by 66PDFcodeScholar
2020

Adaptive Graph Convolutional Recurrent Network for Traffic Forecasting

NeurIPS 2020poster

Modeling complex spatial and temporal correlations in the correlated time series data is indispensable for understanding the traffic dynamics and predicting the future status of an evolving traffic system. Recent works focus on designing complicated graph neural network architectures to capture shar…