Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Junhong Shen

dblp:256/9575 · DBLP profile ↗
← Back
9ranked-venue papers
6as first author
9since 2021 · last 2025
—ORCID · none

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 9 · 6 first-author · 9 since 2021Graphics, computer vision, multimedia, augmented reality and games · 1 · 1 first-author · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
8 papers
Efficient and distributed learning · 25% Transfer learning and domain adaptation · 23% Reinforcement learning · 20%
Computer architecture, parallel and distributed computing, and storage systems
1 paper
Performance modeling and evaluation · 100%

Topics — the 20 heaviest of 22, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Machine learning › Efficient and distributed learning › automated machine learning
neural architecture search
1.122022
NAS-Bench-360: Benchmarking Neural Architecture Search on Diverse Tasks · NeurIPS 2022
Efficient Architecture Search for Diverse Tasks · NeurIPS 2022
Machine learning › Deep learning architectures and training
foundation model
0.912025
Specialized Foundation Models Struggle to Beat Supervised Baselines · ICLR 2025
Machine learning › Transfer learning and domain adaptation › model adaptation
domain specialization
0.812024
Tag-LLM: Repurposing General-Purpose LLMs for Specialized Domains · ICML 2024
Natural language and speech › Language models and text generation › large language model
large language model adaptation
0.812024
Tag-LLM: Repurposing General-Purpose LLMs for Specialized Domains · ICML 2024
Machine learning › Efficient and distributed learning
parameter-efficient fine-tuning
0.812024
Tag-LLM: Repurposing General-Purpose LLMs for Specialized Domains · ICML 2024
Machine learning › Transfer learning and domain adaptation › fine-tuning
cross-modal fine-tuning
0.712023
Cross-Modal Fine-Tuning: Align then Refine · ICML 2023
Machine learning › Transfer learning and domain adaptation › foundation model adaptation
pre-trained model adaptation
0.712023
Cross-Modal Fine-Tuning: Align then Refine · ICML 2023
Machine learning › Efficient and distributed learning › automated machine learning › neural architecture search › one-shot neural architecture search
differentiable architecture search
0.612022
Efficient Architecture Search for Diverse Tasks · NeurIPS 2022
Performance modeling and evaluation
benchmarking
0.612022
NAS-Bench-360: Benchmarking Neural Architecture Search on Diverse Tasks · NeurIPS 2022
Performance modeling and evaluation › benchmarking › machine learning benchmarking
neural architecture search benchmarks
0.612022
NAS-Bench-360: Benchmarking Neural Architecture Search on Diverse Tasks · NeurIPS 2022
Machine learning › Reinforcement learning
deep reinforcement learning
0.512021
Theoretically Principled Deep RL Acceleration via Nearest Neighbor Function Approximation · AAAI 2021
Machine learning › Optimization for machine learning
gradient-based optimization
0.512021
Iterative Teacher-Aware Learning · NeurIPS 2021
Machine learning › Reinforcement learning › imitation learning
inverse reinforcement learning
0.512021
Iterative Teacher-Aware Learning · NeurIPS 2021
Machine learning › Reinforcement learning
sample efficiency
0.512021
Theoretically Principled Deep RL Acceleration via Nearest Neighbor Function Approximation · AAAI 2021
Machine learning › Reinforcement learning
value function approximation
0.512021
Theoretically Principled Deep RL Acceleration via Nearest Neighbor Function Approximation · AAAI 2021
Natural language and speech › Language models and text generation
LLM agents
0.312025
Thinking vs. Doing: Improving Agent Reasoning by Scaling Test-Time Interaction · NeurIPS 2025
Machine learning › Learning paradigms
supervised learning
0.312025
Specialized Foundation Models Struggle to Beat Supervised Baselines · ICLR 2025
Natural language and speech › Language models and text generation › LLM agents
web agents
0.312025
Thinking vs. Doing: Improving Agent Reasoning by Scaling Test-Time Interaction · NeurIPS 2025
Bioinformatics and computational biology
molecular property prediction
0.212024
Tag-LLM: Repurposing General-Purpose LLMs for Specialized Domains · ICML 2024
Machine learning › Learning theory › computational learning theory
machine teaching
0.112021
Iterative Teacher-Aware Learning · NeurIPS 2021

Methods — techniques the papers use, named apart from their topics

fine-tuning · 1.5meta-learning · 1.5continuous prompt tuning · 1.5reinforcement learning · 0.9pre-training · 0.9hyperparameter tuning · 0.9curriculum learning · 0.9soft prompts · 0.8soft prompt · 0.8embedding alignment · 0.7AutoML · 0.7convolutional neural network · 0.6architecture search · 0.6
YearPublicationVenuePosition
2025 Specialized Foundation Models Struggle to Beat Supervised Baselines
abstract
Following its success for vision and text, the "foundation model" (FM) paradigm—pretraining large models on massive data, then fine-tuning on target tasks—has rapidly expanded to domains in the sciences, engineering, healthcare, and beyond. Has this achieved what the original FMs accomplished, i.e. the supplanting of traditional supervised learning in their domains? To answer we look at three modalities—genomics, satellite imaging, and time series—with multiple recent FMs and compare them to a standard supervised learning workflow: model development, hyperparameter tuning, and training, all using only data from the target task. Across these three specialized domains, we find that it is consistently possible to train simple supervised models—no more complicated than a lightly modified wide ResNet or UNet—that match or even outperform the latest foundation models. Our work demonstrates that the benefits of large-scale pretraining have yet to be realized in many specialized areas, reinforces the need to compare new FMs to strong, well-tuned baselines, and introduces two new, easy-to-use, open-source, and automated workflows for doing so.
Zongzhe Xu, Ritvik Gupta, Wenduo Cheng, Alexander Shen 0003, Junhong Shen, Ameet Talwalkar, Mikhail Khodak
ICLR5
2025 Thinking vs. Doing: Improving Agent Reasoning by Scaling Test-Time Interaction
abstract
Test-time scaling in agentic tasks often relies on generating long reasoning traces ("think" more) before acting, but this does not allow agents to acquire new information from the environment or adapt behavior over time. In this work, we propose scaling test-time interaction, an untapped dimension for test-time scaling that increases the agent's interaction horizon to enable rich behaviors such as exploration, backtracking, and dynamic re-planning within a single rollout. To demonstrate the promise of this scaling dimension, we situate our study in the domain of web agents. We first show that even prompting-based interaction scaling can improve task success on web benchmarks non-trivially. Building on this, we introduce TTI, a curriculum-based online reinforcement learning (RL) approach that trains agents by adaptively adjusting their interaction lengths during rollout. Using a Gemma 3 12B model, TTI sets a new state-of-the-art among open-source agents trained on public data on WebVoyager and WebArena. Case studies further reveal that TTI enables agents to balance exploration and exploitation adaptively. Our results establish interaction scaling as a powerful, complementary axis to scaling per-action compute, offering new avenues for training robust and adaptive agents.
Junhong Shen, Lunjun Zhang, Amrith Setlur, Peter Tong, Diego Caples, Nan Jiang 0008, Tong Zhang 0001, Ameet Talwalkar, Aviral Kumar
NeurIPS1
2025 CAT: Content-Adaptive Image Tokenization
abstract
Most existing image tokenizers encode images into a fixed number of tokens or patches, overlooking the inherent variability in image complexity and introducing unnecessary computate overhead for simpler images. To address this, we propose Content-Adaptive Tokenizer (CAT), which dynamically adjusts representation capacity based on the image content and encodes simpler images into fewer tokens. We design (1) a caption-based evaluation system that leverages LLMs to predict content complexity and determine the optimal compression ratio for an image, and (2) a novel nested VAE architecture that performs variable-rate compression in a single model. Trained on images with varying complexity, CAT achieves an average of 15% reduction in rFID across seven detail-rich datasets containing text, humans, and complex textures. On natural image datasets like ImageNet and COCO, it reduces token usage by 18% while maintaining high-fidelity reconstructions. We further evaluate CAT on two downstream tasks. For image classification, CAT consistently improves top-1 accuracy across five datasets spanning diverse domains. For image generation, it boosts training throughput by 23% on ImageNet, leading to more efficient learning and improved FIDs over fixed-token baselines.
Junhong Shen, Kushal Tirumala, Michihiro Yasunaga, Ishan Misra, Luke Zettlemoyer, Lili Yu, Chunting Zhou
NeurIPS1
2024 Tag-LLM: Repurposing General-Purpose LLMs for Specialized Domains
abstract
Large Language Models (LLMs) have demonstrated remarkable proficiency in understanding and generating natural language. However, their capabilities wane in highly specialized domains underrepresented in the pretraining corpus, such as physical and biomedical sciences. This work explores how to repurpose general LLMs into effective task solvers for specialized domains. We introduce a novel, model-agnostic framework for learning custom input tags, which are parameterized as continuous vectors appended to the LLM’s embedding layer, to condition the LLM. We design two types of input tags: domain tags are used to delimit specialized representations (e.g., chemical formulas) and provide domain-relevant context; function tags are used to represent specific functions (e.g., predicting molecular properties) and compress function-solving instructions. We develop a three-stage protocol to learn these tags using auxiliary data and domain knowledge. By explicitly disentangling task domains from task functions, our method enables zero-shot generalization to unseen problems through diverse combinations of the input tags. It also boosts LLM’s performance in various specialized domains, such as predicting protein or chemical properties and modeling drug-target interactions, outperforming expert models tailored to these tasks.
Junhong Shen, Neil A. Tenenholtz, James Brian Hall, David Alvarez-Melis, Nicolò Fusi
ICML1
2023 Cross-Modal Fine-Tuning: Align then Refine
abstract
Fine-tuning large-scale pretrained models has led to tremendous progress in well-studied modalities such as vision and NLP. However, similar gains have not been observed in many other modalities due to a lack of relevant pretrained models. In this work, we propose ORCA, a general cross-modal fine-tuning framework that extends the applicability of a single large-scale pretrained model to diverse modalities. ORCA adapts to a target task via an align-then-refine workflow: given the target input, ORCA first learns an embedding network that aligns the embedded feature distribution with the pretraining modality. The pretrained model is then fine-tuned on the embedded data to exploit the knowledge shared across modalities. Through extensive experiments, we show that ORCA obtains state-of-the-art results on 3 benchmarks containing over 60 datasets from 12 modalities, outperforming a wide range of hand-designed, AutoML, general-purpose, and task-specific cross-modal methods. We highlight the importance of data alignment via a series of ablation studies and exemplify ORCA's utility in data-limited regimes.
Junhong Shen, Liam Li, Lucio M. Dery, Corey Staten, Mikhail Khodak, Graham Neubig, Ameet Talwalkar
ICML1
2022 Efficient Architecture Search for Diverse Tasks
abstract
While neural architecture search (NAS) has enabled automated machine learning (AutoML) for well-researched areas, its application to tasks beyond computer vision is still under-explored. As less-studied domains are precisely those where we expect AutoML to have the greatest impact, in this work we study NAS for efficiently solving diverse problems. Seeking an approach that is fast, simple, and broadly applicable, we fix a standard convolutional network (CNN) topology and propose to search for the right kernel sizes and dilations its operations should take on. This dramatically expands the model's capacity to extract features at multiple resolutions for different types of data while only requiring search over the operation space. To overcome the efficiency challenges of naive weight-sharing in this search space, we introduce DASH, a differentiable NAS algorithm that computes the mixture-of-operations using the Fourier diagonalization of convolution, achieving both a better asymptotic complexity and an up-to-10x search time speedup in practice. We evaluate DASH on ten tasks spanning a variety of application domains such as PDE solving, protein folding, and heart disease detection. DASH outperforms state-of-the-art AutoML methods in aggregate, attaining the best-known automated performance on seven tasks. Meanwhile, on six of the ten tasks, the combined search and retraining time is less than 2x slower than simply training a CNN backbone that is far less accurate.
Junhong Shen, Mikhail Khodak, Ameet Talwalkar
NeurIPS1
2022 NAS-Bench-360: Benchmarking Neural Architecture Search on Diverse Tasks
abstract
Most existing neural architecture search (NAS) benchmarks and algorithms prioritize well-studied tasks, e.g. image classification on CIFAR or ImageNet. This makes the performance of NAS approaches in more diverse areas poorly understood. In this paper, we present NAS-Bench-360, a benchmark suite to evaluate methods on domains beyond those traditionally studied in architecture search, and use it to address the following question: do state-of-the-art NAS methods perform well on diverse tasks? To construct the benchmark, we curate ten tasks spanning a diverse array of application domains, dataset sizes, problem dimensionalities, and learning objectives. Each task is carefully chosen to interoperate with modern CNN-based search methods while possibly being far-afield from its original development domain. To speed up and reduce the cost of NAS research, for two of the tasks we release the precomputed performance of 15,625 architectures comprising a standard CNN search space. Experimentally, we show the need for more robust NAS evaluation of the kind NAS-Bench-360 enables by showing that several modern NAS procedures perform inconsistently across the ten tasks, with many catastrophically poor results. We also demonstrate how NAS-Bench-360 and its associated precomputed results will enable future scientific discoveries by testing whether several recent hypotheses promoted in the NAS literature hold on diverse tasks. NAS-Bench-360 is hosted at https://nb360.ml.cmu.edu.
Renbo Tu, Nicholas Roberts, Mikhail Khodak, Junhong Shen, Frederic Sala, Ameet Talwalkar
NeurIPS4
2021 Theoretically Principled Deep RL Acceleration via Nearest Neighbor Function Approximation
abstract
Recently, deep reinforcement learning (RL) has achieved remarkable empirical success by integrating deep neural networks into RL frameworks. However, these algorithms often require a large number of training samples and admit little theoretical understanding. To mitigate these issues, we propose a theoretically principled nearest neighbor (NN) function approximator that can replace the value networks in deep RL methods. Inspired by human similarity judgments, the NN approximator estimates the action values using rollouts on past observations and can provably obtain a small regret bound that depends only on the intrinsic complexity of the environment. We present (1) Nearest Neighbor Actor-Critic (NNAC), an online policy gradient algorithm that demonstrates the practicality of combining function approximation with deep RL, and (2) a plug-and-play NN update module that aids the training of existing deep RL methods. Experiments on classical control and MuJoCo locomotion tasks show that the NN-accelerated agents achieve higher sample efficiency and stability than the baseline agents. Based on its theoretical benefits, we believe that the NN approximator can be further applied to other complex domains to speed-up learning.
Junhong Shen, Lin Yang 0011
AAAI1
2021 Iterative Teacher-Aware Learning
abstract
In human pedagogy, teachers and students can interact adaptively to maximize communication efficiency. The teacher adjusts her teaching method for different students, and the student, after getting familiar with the teacher’s instruction mechanism, can infer the teacher’s intention to learn faster. Recently, the benefits of integrating this cooperative pedagogy into machine concept learning in discrete spaces have been proved by multiple works. However, how cooperative pedagogy can facilitate machine parameter learning hasn’t been thoroughly studied. In this paper, we propose a gradient optimization based teacher-aware learner who can incorporate teacher’s cooperative intention into the likelihood function and learn provably faster compared with the naive learning algorithms used in previous machine teaching works. We give theoretical proof that the iterative teacher-aware learning (ITAL) process leads to local and global improvements. We then validate our algorithms with extensive experiments on various tasks including regression, classification, and inverse reinforcement learning using synthetic and real data. We also show the advantage of modeling teacher-awareness when agents are learning from human teachers.
Luyao Yuan, Dongruo Zhou, Junhong Shen, Jingdong Gao, Jeffrey L. Chen, Quanquan Gu, Ying Nian Wu, Song-Chun Zhu
NeurIPS3