← Search

Shihao Bai

7 accepted papers

2026

StructureBench: A Unified Benchmark Suite for Multi-Scenario Structured Generation Tasks with On-Device Models

IJCAI 2026

Structured output generation is increasingly critical for real-world AI systems, particularly in on-device settings where small language models (0.5B–8B parameters) must produce machine-executable outputs under strict latency and privacy constraints. Although constrained decoding provides formal gua

Cited by 0Scholar
2025

AtomNet: Designing Tiny Models from Operators Under Extreme MCU Constraints

AAAI 2025technical

Tiny machine learning (TinyML) has attracted heightened attention for its ability to provide low-cost and instantaneous performance on edge devices. Particularly, the commonly used microcontroller unit (MCU) imposes extreme constraints on peak memory (SRAM) and storage (Flash). Existing TinyML metho…

Cited by 0SourcePDFScholar
2025

Pre3: Enabling Deterministic Pushdown Automata for Faster Structured LLM Generation

ACL 2025long

Extensive LLM applications demand efficient structured generations, particularly for LR(1) grammars, to produce outputs in specified formats (e.g., JSON). Existing methods primarily parse LR(1) grammars into a pushdown automaton (PDA), leading to runtime execution overhead for context-dependent toke…

2023

Lossy and Lossless (L2) Post-training Model Size Compression

ICCV 2023poster

Deep neural networks have delivered remarkable performance and have been widely used in various visual tasks. However, their huge sizes cause significant inconvenience for transmission and storage. Many previous studies have explored model size compression. However, these studies often approach vari…

Cited by 6PDFcodeScholar
2021

Stratified Rule-Aware Network for Abstract Visual Reasoning

AAAI 2021technical

Abstract reasoning refers to the ability to analyze information, discover rules at an intangible level, and solve problems in innovative ways. Raven's Progressive Matrices (RPM) test is typically used to examine the capability of abstract reasoning. The subject is asked to identify the correct choic…

2020

Few-shot Visual Learning with Contextual Memory and Fine-grained Calibration

IJCAI 2020poster

Few-shot learning aims to learn a model that can be readily adapted to new unseen classes (concepts) by accessing one or few examples. Despite the successful progress, most of the few-shot learning approaches, concentrating on either global or local characteristics of examples, still suffer from wea…

Cited by 0SourcePDFScholar
2020

Transductive Relation-Propagation Network for Few-shot Learning

IJCAI 2020poster

Few-shot learning, aiming to learn novel concepts from few labeled examples, is an interesting and very challenging problem with many practical advantages. To accomplish this task, one should concentrate on revealing the accurate relations of the support-query pairs. We propose a transductive relati…

Cited by 0SourcePDFScholar