← Search

Tao Lei

25 accepted papers

2026

Adaptive Thinking: Large Language Models Know When to Think in Latent Space

ICLR 2026poster

Recent advances in large language models (LLMs) test-time computing have introduced the capability to perform intermediate chain-of-thought (CoT) reasoning (thinking) before generating answers. While increasing the thinking budget yields smooth performance improvements at inference time, the relatio…

Cited by 0SourceScholar
2026

DGKAN: Dual-branch Graph Kolmogorov-Arnold Network for Unsupervised Multimodal Change Detection

AAAI 2026technical

Multimodal change detection (MCD) has important applications in disaster assessment, but the nonlinear distortion of features and spatial misalignment caused by sensor imaging differences make it difficult to obtain changes through direct comparison. To overcome the above problems, this study aims t

Cited by 0SourcePDFScholar
2026

SPEGC: Continual Test-Time Adaptation via Semantic-Prompt-Enhanced Graph Clustering for Medical Image Segmentation

CVPR 2026

In medical image segmentation tasks, the domain gap caused by the difference in data collection between training and testing data seriously hinders the deployment of pre-trained models in clinical practice. Continual Test-Time Adaptation (CTTA) aims to enable pre-trained models to adapt to continuou

Cited by 0SourcecodeScholar
2025

Adaptive Learning of High-Value Regions for Semi-Supervised Medical Image Segmentation

ICCV 2025poster

Existing semi-supervised learning methods typically mitigate the impact of unreliable predictions by suppressing low-confidence regions. However, these methods fail to explore which regions hold higher learning value and how to design adaptive learning strategies for these regions. To address these…

2025

Dynamic Sparse Encoding and Cross-Temporal Attention for Remote Sensing Image Change Detection

ICASSP 2025accepted

Due to the inherent inductive bias of operations, convolutional neural networks (CNN) cannot model global information of remote sensing (RS) images. In contrast, Transformer-based methods can establish long-range dependencies of images through self-attention (SA) mechanism, but it faces the challeng…

Cited by 1SourceScholar
2025

EC-DIT: Scaling Diffusion Transformers with Adaptive Expert-Choice Routing

ICLR 2025poster

Diffusion transformers have been widely adopted for text-to-image synthesis. While scaling these models up to billions of parameters shows promise, the effectiveness of scaling beyond current sizes remains underexplored and challenging. By explicitly exploiting the computational heterogeneity of ima…

Cited by 0SourcePDFScholar
2025

Instruction-Following Pruning for Large Language Models

ICML 2025poster

With the rapid scaling of large language models (LLMs), structured pruning has become a widely used technique to learn efficient, smaller models from larger ones, delivering superior performance compared to training similarly sized models from scratch. In this paper, we move beyond the traditional s…

Cited by 0SourcePDFScholar
2024

"MM1: Methods, Analysis & Insights from Multimodal LLM Pre-training"

ECCV 2024poster

"In this work, we discuss building performant Multimodal Large Language Models (MLLMs). In particular, we study the importance of various architecture components and data choices. Through careful and comprehensive ablations of the image encoder, the vision language connector, and various pre-trainin…

2023

CiT-Net: Convolutional Neural Networks Hand in Hand with Vision Transformers for Medical Image Segmentation

IJCAI 2023poster

The hybrid architecture of convolutional neural networks (CNNs) and Transformer are very popular for medical image segmentation. However, it suffers from two challenges. First, although a CNNs branch can capture the local image features using vanilla convolution, it cannot achieve adaptive feature l…

2023

CoLT5: Faster Long-Range Transformers with Conditional Computation

EMNLP 2023long main

Many natural language processing tasks benefit from long inputs, but processing long documents with Transformers is expensive -- not only due to quadratic attention complexity but also from applying feedforward and projection layers to every token. However, not all tokens are equally important, espe…

Cited by 0SourceScholar
2023

Conditional Adapters: Parameter-efficient Transfer Learning with Fast Inference

NeurIPS 2023poster

We propose Conditional Adapter (CoDA), a parameter-efficient transfer learning method that also improves inference efficiency. CoDA generalizes beyond standard adapter approaches to enable a new way of balancing speed and accuracy using conditional computation. Starting with an existing dense pretra…

Cited by 63SourcePDFScholar
2023

Local-Global Siamese Network with Efficient Inter-Scale Feature Learning for Change Detection in VHR Remote Sensing Images

ICASSP 2023accepted

The popular networks for change detection (CD) in very-high-resolution (VHR) remote sensing (RS) images usually suffer from two problems. First, it is difficult for these networks to model simultaneously the local and global features of changed targets, which leads to the limited feature representat…

Cited by 1SourceScholar
2023

Rethinking the Role of Token Retrieval in Multi-Vector Retrieval

NeurIPS 2023poster

Multi-vector retrieval models such as ColBERT [Khattab et al., 2020] allow token-level interactions between queries and documents, and hence achieve state of the art on many information retrieval benchmarks. However, their non-linear scoring function cannot be scaled to millions of documents, necess…

2022

Global Evolution Neural Network for Segmentation of Remote Sensing Images

ICASSP 2022accepted

The popular convolutional neural networks (CNNs) have been successfully used in very high-resolution remote sensing image semantic segmentation. However, these networks often suffer from performance limitations. First, although deeper networks usually provide better feature representation, they may…

Cited by 0SourceScholar
2022

Mixture-of-Experts with Expert Choice Routing

NeurIPS 2022accept

Sparsely-activated Mixture-of-experts (MoE) models allow the number of parameters to greatly increase while keeping the amount of computation for a given token or a given sample unchanged. However, a poor expert routing strategy (e.g. one resulting in load imbalance) can cause certain experts to be…

Cited by 336SourcePDFScholar
2022

SRU++: Pioneering Fast Recurrence with Attention for Speech Recognition

ICASSP 2022accepted

The Transformer architecture has been well adopted as a dominant architecture in most sequence transduction tasks including automatic speech recognition (ASR), since its attention mechanism excels in capturing long-range dependencies. While models built solely upon attention can be better paralleliz…

Cited by 0SourceScholar
2021

Lightweight Non-Local Network for Image Super-Resolution

ICASSP 2021accepted

The popular deep convolutional networks used for image super-resolution (SR) reconstruction often increase the network depth and employ attention mechanism to improve image reconstruction effect. However, these networks suffer from two problems. The first is the deeper network easily causes higher c…

Cited by 0SourceScholar
2021

Nutri-bullets: Summarizing Health Studies by Composing Segments

AAAI 2021technical

We introduce Nutri-bullets, a multi-document summarization task for health and nutrition. First, we present two datasets of food and health summaries from multiple scientific studies. Furthermore, we propose a novel extract-compose model to solve the problem in the regime of limited parallel data. W…

2020

Lightweight V-Net for Liver Segmentation

ICASSP 2020accepted

The V-Net based 3D fully convolutional neural networks have been widely used in liver volumetric data segmentation. However, due to the large number of parameters of these networks, 3D FCNs suffer from high computational cost and GPU memory usage. To address these issues, we design a lightweight V-N…

Cited by 0SourceScholar
2019

End-to-end Change Detection Using a Symmetric Fully Convolutional Network for Landslide Mapping

ICASSP 2019accepted

In this paper, we propose a novel approach based on a symmetric fully convolutional network within pyramid pooling (FCN-PP) for landslide mapping (LM). The proposed approach has three advantages. Firstly, this approach is automatic and insensitive to noise because multivariate morphological reconstr…

Cited by 0SourceScholar
2017

Deriving Neural Architectures from Sequence and Graph Kernels

ICML 2017poster

The design of neural architectures for structured objects is typically guided by experimental insights rather than a formal process. In this work, we appeal to kernels over combinatorial structures, such as sequences and graphs, to derive appropriate neural operations. We introduce a class of deep r…

2017

Style Transfer from Non-Parallel Text by Cross-Alignment

NeurIPS 2017poster

This paper focuses on style transfer on the basis of non-parallel text. This is an instance of a broad family of problems including machine translation, decipherment, and sentiment modification. The key challenge is to separate the content from other aspects such as style. We assume a shared latent…