TOMAGPT

TOMAGPT

A Qwen3-4B-Instruct-2507 model fine-tuned with GRPO (Group Relative Policy Optimization) to classify legal hearsay by decomposing it into three sub-elements under the U.S. Federal Rules of Evidence.

What It Does

TOMAGPT classifies whether a statement is hearsay by analyzing three sub-elements:

1. Assertion -- Is the statement an assertion?
2. Out-of-court -- Was the statement made out of court?
3. TOMA -- Is the statement offered to prove the truth of the matter asserted?

Hearsay = YES only if all three sub-elements are YES.

Results

Evaluated on the LegalBench hearsay test set (94 examples):

Training Details

• Method: GRPO (Group Relative Policy Optimization)
• Platform: Prime Intellect Lab
• Environment: smolclaims/TOMAGPT (v0.3.0)
• Base model: Qwen/Qwen3-4B-Instruct-2507
• Training data: DoodDood/HearsayGRPOTrainingData2 (3,140 examples)
• Steps: 500
• Learning rate: 1e-5
• Batch size: 128
• Rollouts per example: 16

LoRA Configuration

• Rank (r): 16
• Alpha: 32
• Dropout: 0.0
• Target modules: q_proj, k_proj, v_proj, o_proj, gate_proj, up_proj, down_proj

Reward Functions

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model = AutoModelForCausalLM.from_pretrained(
    "DoodDood/TOMAGPT", torch_dtype=torch.bfloat16, device_map="auto")
tokenizer = AutoTokenizer.from_pretrained("DoodDood/TOMAGPT")

system_prompt = (
    "You are a legal assistant identifying hearsay. Hearsay is defined as "
    "an out-of-court statement introduced to prove the truth of the matter "
    "asserted.\n\n"
    "Respond in EXACTLY this format (semicolon-separated):\n"
    "is_hearsay: YES/NO; an_assertion: YES/NO; made_out_of_court: YES/NO; "
    "is_for_toma: YES/NO"
)

scenario = "At trial, the prosecution presents testimony from a police officer who states that a bystander at the scene told him, 'The defendant ran the red light.'"

messages = [
    {"role": "system", "content": system_prompt},
    {"role": "user", "content": scenario}
]

text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
inputs = tokenizer([text], return_tensors="pt").to(model.device)

with torch.no_grad():
    output = model.generate(**inputs, max_new_tokens=128, do_sample=False)

response = tokenizer.decode(output[0][len(inputs.input_ids[0]):], skip_special_tokens=True)
print(response)
# Expected: is_hearsay: YES; an_assertion: YES; made_out_of_court: YES; is_for_toma: YES

Links

• Training data: DoodDood/HearsayGRPOTrainingData2
• GRPO environment: smolclaims/TOMAGPT on Prime Intellect
• Eval benchmark: nguha/legalbench (hearsay subset)