Model Card: Llama-3.2-3B-Korean-NL2SQL (Merged)

๋ณธ ๋ชจ๋ธ์€ Meta์˜ Llama-3.2-3B-Instruct ๋ชจ๋ธ์„ ๋ฒ ์ด์Šค๋กœ ํ•˜์—ฌ, ์‹ค๋ฌด์šฉ PostgreSQL ํ™˜๊ฒฝ์—์„œ ํ•œ๊ตญ์–ด ์งˆ์˜๋ฅผ ๊ณ ์ •๋ฐ€ SQL ์ฟผ๋ฆฌ๋กœ ๋ณ€ํ™˜(NL2SQL)ํ•  ์ˆ˜ ์žˆ๋„๋ก ํŒŒ์ธํŠœ๋‹ ๋ฐ ๊ฐ€์ค‘์น˜ ๋ณ‘ํ•ฉ(Weight Merge)์„ ์™„๋ฃŒํ•œ ๊ฒฝ๋Ÿ‰ ํŠนํ™” ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค.

3B๋ผ๋Š” ์†Œํ˜• ์ฒด๊ธ‰์ž„์—๋„ ๋ถˆ๊ตฌํ•˜๊ณ , ์—๋Ÿฌ ํ”ผ๋“œ๋ฐฑ์„ ๊ธฐ๋ฐ˜์œผ๋กœ ํ•œ ์žฌ๊ท€์  ์ž๊ฐ€ ์ˆ˜์ • ์—์ด์ „ํŠธ ํ™˜๊ฒฝ์—์„œ 22.73%๋ผ๋Š” ๋†’์€ ๋ณต๊ตฌ ์„ฑ๊ณต๋ฅ ์„ ๊ธฐ๋กํ•˜๋ฉฐ ๋›ฐ์–ด๋‚œ ์œ ์—ฐ์„ฑ๊ณผ ๋น„์šฉ ํšจ์œจ์„ฑ(Cost-Efficiency)์„ ์ฆ๋ช…ํ•œ ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค.

๐ŸŒŸ ์ฃผ์š” ํŠน์ง• (Key Features)

  • ์ดˆ๊ฒฝ๋Ÿ‰ยท๊ณ ํšจ์œจ ์—”์ง€๋‹ˆ์–ด๋ง: 3B ํŒŒ๋ผ๋ฏธํ„ฐ ์‚ฌ์ด์ฆˆ๋กœ VRAM ์†Œ๋ชจ๋ฅผ ์ตœ์†Œํ™”ํ•˜์—ฌ, ์‚ฌ์–‘์ด ์ œํ•œ๋œ ์—ฃ์ง€ ์„œ๋ฒ„๋‚˜ ๋กœ์ปฌ ํ™˜๊ฒฝ์—์„œ๋„ ๋Œ€๊ทœ๋ชจ ์ธํ”„๋ผ ๋ถ€๋‹ด ์—†์ด ๊ณ ์† ์ถ”๋ก  ๋ฐ ์„œ๋น™์ด ๊ฐ€๋Šฅํ•ฉ๋‹ˆ๋‹ค.
  • ๋†’์€ ํšŒ๋ณต ํƒ„๋ ฅ์„ฑ (High Resilience): ์ตœ์ดˆ ์ƒ์„ฑ์—์„œ ๋ฌธ๋ฒ•์  ์‹ค์ˆ˜๊ฐ€ ๋ฐœ์ƒํ•˜๋”๋ผ๋„, ๋ฐ์ดํ„ฐ๋ฒ ์ด์Šค ์—๋Ÿฌ ๋กœ๊ทธ๋ฅผ ์ฃผ์ž…ํ–ˆ์„ ๋•Œ ๋ฌธ๋งฅ์„ ํŒŒ์•…ํ•˜์—ฌ ์˜ฌ๋ฐ”๋ฅธ ์ฟผ๋ฆฌ๋กœ ๊ณ ์ณ ์“ฐ๋Š” ๋””๋ฒ„๊น… ๋Šฅ๋ ฅ์ด ์ฒด๊ธ‰ ๋Œ€๋น„ ๋งค์šฐ ํƒ์›”ํ•ฉ๋‹ˆ๋‹ค.
  • ์‹ค๋ฌดํ˜• PostgreSQL ๋งคํ•‘: ์†Œํ˜• ๋ชจ๋ธ์ด ํ”ํžˆ ๋ฒ”ํ•˜๊ธฐ ์‰ฌ์šด ๋ฌธ๋ฒ•์  ๋น„์•ฝ์„ ์–ต์ œํ•˜๊ณ , PostgreSQL ๊ทœ๊ฒฉ์— ๋งž๋Š” ์•ˆ์ •์ ์ธ ์ฟผ๋ฆฌ ํŒจํ„ด์„ ๊ตฌ์‚ฌํ•˜๋„๋ก ํŠœ๋‹๋˜์—ˆ์Šต๋‹ˆ๋‹ค.

๐Ÿ“Š ์„ฑ๋Šฅ ํ‰๊ฐ€ ์š”์•ฝ (Evaluation Results)

์‹ค๋ฌด์šฉ ERP ๋ฐ์ดํ„ฐ๋ฒ ์ด์Šค ์Šคํ‚ค๋งˆ์™€ ๋‚œ์ด๋„๋ณ„(Level 1 ~ 5) ํ‰๊ฐ€ ๋ฐ์ดํ„ฐ์…‹ 400๋ฌธํ•ญ์„ ๋ฐ”ํƒ•์œผ๋กœ ์—„๋ฐ€ํ•˜๊ฒŒ ์ธก์ •ํ•œ ๋ฒค์น˜๋งˆํฌ ๊ฒฐ๊ณผ์ž…๋‹ˆ๋‹ค.

Difficulty Pure Acc Final Acc Errors Repaired Repair Rate
Level 1 92.50% 92.50% 1 0 0.00%
Level 2 85.00% 87.50% 5 2 40.00%
Level 3 76.25% 78.75% 3 2 66.67%
Level 4 60.00% 61.25% 6 1 16.67%
Level 5 53.75% 53.75% 7 0 0.00%
TOTAL 73.50% 74.75% 22 5 22.73%
  • Pure Accuracy: ์ตœ์ดˆ 1ํšŒ ์ƒ์„ฑ ์‹œ์˜ SQL ์‹คํ–‰ ๊ฒฐ๊ณผ ์ •๋‹ต๋ฅ ์€ **73.50%**์ž…๋‹ˆ๋‹ค.
  • ์—์ด์ „ํŠธ ์ž๊ฐ€ ์ˆ˜์ • ๊ธฐ์—ฌ๋„: 1์ฐจ ์ƒ์„ฑ์—์„œ ์—๋Ÿฌ๊ฐ€ ๋ฐœ์ƒํ•œ 22๊ฑด ์ค‘ 5๊ฑด์„ ์Šค์Šค๋กœ ์™„๋ฒฝํžˆ ์ˆ˜๋ฆฌ(Repair Success Rate: 22.73%)ํ•ด๋‚ด๋ฉฐ, ์ตœ์ข… ์ •๋‹ต๋ฅ ์„ **74.75%**๊นŒ์ง€ ๋Œ์–ด์˜ฌ๋ ธ์Šต๋‹ˆ๋‹ค. ํŠนํžˆ ์ค‘๊ฐ„ ๋‚œ์ด๋„(Level 2, 3)์—์„œ ์ตœ๋Œ€ 66.67%์˜ ๊ณ ํšจ์œจ ๋ณต๊ตฌ ์„ฑ๋Šฅ์„ ์ž…์ฆํ–ˆ์Šต๋‹ˆ๋‹ค.

๐Ÿ’ป ์‚ฌ์šฉ ๋ฐฉ๋ฒ• (How to Use)

Llama-3.2์˜ ๊ณต์‹ Chat Template ๊ทœ๊ฒฉ์„ ์ค€์ˆ˜ํ•˜์—ฌ ๋กœ์ปฌ ํ™˜๊ฒฝ์—์„œ ์ถ”๋ก ํ•˜๋Š” ์˜ˆ์‹œ ์ฝ”๋“œ์ž…๋‹ˆ๋‹ค.

import torch
from transformers import AutoTokenizer, AutoModelForCausalLM

model_id = "yeongseok11/Llama-3.2-3B-korean-nl2sql"

tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
    model_id,
    torch_dtype=torch.bfloat16,
    device_map="auto"
)
model.eval()

# Prompt Template (Llama-3.2 ์ „์šฉ ํ…œํ”Œ๋ฆฟ ์ค€์ˆ˜)
prompt = """<|begin_of_text|><|start_header_id|>system<|end_header_id|>

๋‹น์‹ ์€ ์‹ค๋ฌด์šฉ PostgreSQL ์ „๋ฌธ๊ฐ€์ž…๋‹ˆ๋‹ค. ์˜ค์ง SQL ์ฟผ๋ฆฌ๋งŒ ๋‹ต๋ณ€ํ•˜์„ธ์š”.<|eot_id|><|start_header_id|>user<|end_header_id|>

### ์Šคํ‚ค๋งˆ:
CREATE TABLE emp (
    emp_id INT PRIMARY KEY,
    emp_name VARCHAR(50),
    dept_id INT,
    salary INT
);

### ์งˆ๋ฌธ:
๊ธฐํšํŒ€(dept_id = 10) ์ง์›๋“ค์˜ ํ‰๊ท  ๊ธ‰์—ฌ๋ฅผ ๊ตฌํ•˜๋Š” ์ฟผ๋ฆฌ๋ฅผ ์งœ์ค˜.<|eot_id|><|start_header_id|>assistant<|end_header_id|>

### SQL:
"""

inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
with torch.no_grad():
    outputs = model.generate(
        **inputs,
        max_new_tokens=256,
        temperature=0.0,
        do_sample=False,
        pad_token_id=tokenizer.eos_token_id
    )

print(tokenizer.decode(outputs[0], skip_special_tokens=True).split("### SQL:\n")[-1])

๐Ÿ“ ์—ฐ๊ตฌ ๋ฐ ํ•œ๊ณ„์  (Limitations & Future Work)
๋ณธ ๋ชจ๋ธ์€ ๊ฒฝ๋Ÿ‰ํ™” ๋ชจ๋ธ๋กœ์„œ ๋›ฐ์–ด๋‚œ ์œ ์—ฐ์„ฑ์„ ๊ฐ–์ถ”๊ณ  ์žˆ์œผ๋‚˜, ์ดˆ๊ณ ๋‚œ๋„(Level 5)์˜ ๋ณต์žกํ•œ ๋‹ค์ค‘ ์กฐ์ธ ์ œ์•ฝ ์กฐ๊ฑด ํ™˜๊ฒฝ์—์„œ๋Š” ๋ชจ๋ธ ์ฒด๊ธ‰ ํ•œ๊ณ„๋กœ ์ธํ•œ ์˜๋ฏธ๋ก ์  ํ•œ๊ณ„๋ฅผ ์ผ๋ถ€ ๋ณด์ž…๋‹ˆ๋‹ค.
 ์ด๋ฅผ ๋ณด์™„ํ•˜๊ธฐ ์œ„ํ•ด ํ–ฅํ›„ ์—ฐ๊ตฌ๋Š” ๊ธฐ์ € ์ธํ”„๋ผ ๋‹จ์—์„œ ์Šคํ‚ค๋งˆ ์ •๋ณด๋ฅผ LLM ์ง€ํ–ฅ์ ์œผ๋กœ ๊ฐ€๊ณตํ•ด ์ฃผ๋Š” AI ์นœํ™”์  ๋ฉ”ํƒ€๋ฐ์ดํ„ฐ ์ž๋™ ๊ด€๋ฆฌ ํŒŒ์ดํ”„๋ผ์ธ(AI-Friendly Metadata Enrichment) ์ฒด๊ณ„์™€์˜ ๊ฒฐํ•ฉ์„ ๋ชฉํ‘œ๋กœ ํ•˜๊ณ  ์žˆ์Šต๋‹ˆ๋‹ค.
Downloads last month
-
Safetensors
Model size
3B params
Tensor type
BF16
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for yeongseok11/Llama-3.2-3B-korean-nl2sql

Adapter
(857)
this model