Papers
arxiv:2503.12730

TinySQL: A Progressive Text-to-SQL Dataset for Mechanistic Interpretability Research

Published on Mar 17
Authors:
,
,
,

Abstract

Text-to-SQL generation is studied using TinySQL to bridge interpretability gaps between simple tasks and large models, employing techniques like Edge Attribution Patching and Sparse Autoencoders to analyze SQL subskills and layerwise logit lens.

AI-generated summary

Mechanistic interpretability research faces a gap between analyzing simple circuits in toy tasks and discovering features in large models. To bridge this gap, we propose text-to-SQL generation as an ideal task to study, as it combines the formal structure of toy tasks with real-world complexity. We introduce TinySQL, a synthetic dataset, progressing from basic to advanced SQL operations, and train models ranging from 33M to 1B parameters to establish a comprehensive testbed for interpretability. We apply multiple complementary interpretability techniques, including Edge Attribution Patching and Sparse Autoencoders, to identify minimal circuits and components supporting SQL generation. We compare circuits for different SQL subskills, evaluating their minimality, reliability, and identifiability. Finally, we conduct a layerwise logit lens analysis to reveal how models compose SQL queries across layers: from intent recognition to schema resolution to structured generation. Our work provides a robust framework for probing and comparing interpretability methods in a structured, progressively complex setting.

Community

Sign up or log in to comment

Models citing this paper 16

Browse 16 models citing this paper

Datasets citing this paper 8

Browse 8 datasets citing this paper

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2503.12730 in a Space README.md to link it from this page.

Collections including this paper 1