Abstract
Schema linking is a critical step in Text-to-SQL task, aiming to accuratelypredict the table names and column names required for the SQL query based onthe given question. However, current fine-tuning approaches for schema linkingmodels employ a rote-learning paradigm, excessively optimizing for ground truthschema linking outcomes while compromising reasoning ability. This limitationarises because of the difficulty in acquiring a high-quality reasoning samplefor downstream tasks. To address this, we propose Schema-R1, a reasoning schemalinking model trained using reinforcement learning. Specifically, Schema-R1consists of three key steps: constructing small batches of high-qualityreasoning samples, supervised fine-tuning for cold-start initialization, andrule-based reinforcement learning training. The final results demonstrate thatour method effectively enhances the reasoning ability of the schema linkingmodel, achieving a 10\% improvement in filter accuracy compared to the existingmethod. Our code is available at https://github.com/hongWin/Schema-R1/.