LLaVA-1.5-7B Skin Disease Fine-tuned Model (LoRA)

Model Description

์ด ๋ชจ๋ธ์€ llava-hf/llava-1.5-7b-hf ๋ฒ ์ด์Šค ๋ชจ๋ธ์„ ๊ธฐ๋ฐ˜์œผ๋กœ ํŒŒ์ธํŠœ๋‹๋œ LoRA ์–ด๋Œ‘ํ„ฐ์ž…๋‹ˆ๋‹ค. ์•ˆ๋ฉด ํ”ผ๋ถ€ ์งˆํ™˜์„ ์ง„๋‹จํ•˜๊ณ , ๊ด€๋ จ ์ผ€์–ด ๊ฐ€์ด๋“œ๋ฅผ ์ œ๊ณตํ•˜๋Š” ๋ฐ ํŠนํ™”๋˜์–ด ์žˆ์Šต๋‹ˆ๋‹ค. ๊ธฐ์กด ๋ฒ”์šฉ ๋ชจ๋ธ์ด ๋™์–‘์ธ(ํ•œ๊ตญ์ธ) ํ”ผ๋ถ€ ์ž„์ƒ ๋ฐ์ดํ„ฐ ํ•™์Šต์ด ๋ถ€์กฑํ•ด ๋ฐœ์ƒํ•˜๋Š” ์˜ค์ง„์œจ๊ณผ ํ• ๋ฃจ์‹œ๋„ค์ด์…˜(์ž„์˜ ์ฒ˜๋ฐฉ)์„ ํ•ด๊ฒฐํ•˜๊ณ ์ž ๊ฐœ๋ฐœ๋˜์—ˆ์Šต๋‹ˆ๋‹ค.

  • Base Model: llava-hf/llava-1.5-7b-hf
  • Finetuning Method: 8-bit QLoRA & RAFT (Retrieval-Augmented Fine-Tuning)
  • Primary Use Case: ์•ˆ๋ฉด ํ”ผ๋ถ€ ์งˆํ™˜ ์ถ”๋ก  ๋ฐ ๋‹ค์ค‘ ํ„ด(Multi-turn) ์งˆ์˜์‘๋‹ต (์ž๊ฐ€ ์ง„๋‹จ ์ฑ—๋ด‡)

Training Details

  • Training Data: AI Hub์˜ '์•ˆ๋ฉด๋ถ€ ํ”ผ๋ถ€์งˆํ™˜ ์ด๋ฏธ์ง€ ํ•ฉ์„ฑ ๋ฐ์ดํ„ฐ' 9,600์žฅ์„ ๊ธฐ๋ฐ˜์œผ๋กœ, ์ด๋ฏธ์ง€ ๋‹น 4๊ฐœ์˜ ๋‹จ์ผ ํ„ด ์งˆ์˜์‘๋‹ต์„ ํ•˜๋‚˜์˜ ๋Œ€ํ™” ์„ธ์…˜์œผ๋กœ ๋ฌถ๋Š” ๋ฉ€ํ‹ฐํ„ด ์„ธ์…˜ ์ฒด์ด๋‹(Multi-turn Session Chaining) ๊ธฐ๋ฒ•์„ ์ ์šฉํ•˜์—ฌ 38,400 Turn ๋Œ€ํ™”์…‹(QA์Œ)์œผ๋กœ ๊ฐ€๊ณตํ–ˆ์Šต๋‹ˆ๋‹ค.
  • Results: ํ”ผ๋ถ€ ์งˆํ™˜ ์ง„๋‹จ Accuracy ์•ฝ 60% ํ–ฅ์ƒ (0.093 โž” 0.148), Macro F1-Score ์•ฝ 65% ํ–ฅ์ƒ (0.126 โž” 0.208). ๋ฐ˜๋ณต ์ƒ์„ฑ ๋ฃจํ”„(Repetition Loop) ๋ฒ„๊ทธ๋ฅผ ๋ฐ์ดํ„ฐ ์ „์ฒ˜๋ฆฌ ๋ ˆ๋ฒจ์—์„œ ์›์ฒœ ํ•ด๊ฒฐํ–ˆ์Šต๋‹ˆ๋‹ค.

How to Get Started with the Model

์ด ๋ชจ๋ธ์€ ์ „์ฒด ๋ชจ๋ธ์ด ์•„๋‹Œ LoRA ์–ด๋Œ‘ํ„ฐ(๊ฐ€์ค‘์น˜) ์ž…๋‹ˆ๋‹ค. ์‚ฌ์šฉํ•˜์‹œ๋ ค๋ฉด peft ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋ฅผ ํ†ตํ•ด ๋ฒ ์ด์Šค ๋ชจ๋ธ์— ์–ด๋Œ‘ํ„ฐ๋ฅผ ๋กœ๋“œํ•ด์•ผ ํ•ฉ๋‹ˆ๋‹ค.

from transformers import AutoProcessor, LlavaForConditionalGeneration
from peft import PeftModel
import torch

# 1. Load Base Model & Processor
base_model_id = "llava-hf/llava-1.5-7b-hf"
processor = AutoProcessor.from_pretrained(base_model_id)
base_model = LlavaForConditionalGeneration.from_pretrained(
    base_model_id,
    torch_dtype=torch.float16,
    device_map="auto"
)

# 2. Load LoRA Adapter
adapter_id = "jun47/llava-7b-skin"
model = PeftModel.from_pretrained(base_model, adapter_id)
Downloads last month
60
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for jun47/llava-7b-skin

Adapter
(152)
this model