VertexAGI/prism-caption-1-5-micro

Prism Caption 1.5 Micro

A 0.6B-parameter model that does one thing: turn a user's first message into a short chat title.

Part of the Prism family of small, single-purpose models. Successor to Prism Caption 1 Micro (Gemma-3-1B, 339 examples).

Overview

Given the first message in a new conversation, this model returns a short, specific title — the "New chat" auto-naming behaviour that ChatGPT, Claude and similar assistants provide. Nothing else. At 0.6B parameters it is small and fast enough to run this narrow task locally instead of spending a frontier-model call on it.

Results

Evaluated on 24 held-out topics that appear nowhere in the training data (the v1 evaluation sampled from its own training topic list, so it measured memorisation rather than generalisation — that mistake is corrected here).

Chat titling has no single correct answer, so the metrics are behavioural:

The win is format reliability. The base model leaks the prompt back ("What's the deal with picking a beginner-friendly board game for si..."), keeps question marks ("Sourdough starter smells like acetone?"), and echoes the user's phrasing instead of titling it. The tuned model does not do these things.

The honest cost: it is terser, and sometimes too terse. Relevance fell from 24/24 to 21/24 because a few titles over-compress and drop the actual subject:

Most titles are good — "Last Minute Flight Cancellation", "First Half Marathon Training", "Laptop Lid Residue Removal", "Car Braking Grinding Noise" — but if your UI needs the subject noun preserved, test on your own traffic first.

Training

Teacher. Titles were distilled through NVIDIA NIM. Generation began on google/diffusiongemma-26b-a4b-it, which produced good titles but whose NIM rate limit is far tighter than the nemotron endpoints — 82% of requests returned 429. The remaining ~957 examples were generated with nvidia/nemotron-3.5-lightning-30b-a3b. The corpus is therefore mixed-teacher: roughly 43 examples from diffusiongemma, the rest from Lightning.

On topic breadth. v1 used 51 topics for 339 examples. Scaling that to 1,000 would mean ~20 repeats per topic; narrow template pools cause the model to memorise the pool instead of learning the task. The bank was widened to 167 topics first, so each recurs ~6 times. The result set contains 714 unique titles across 1,000 examples.

Limitations

• —This is a 0.6B-parameter model — small enough to run cheaply and locally, but limited in general capability outside its one narrow task.
• —Over-compression. Averages 3.3 words against a 3-6 word spec, and occasionally drops the subject noun (see above).
• —English only.
• —First messages only. It was not trained to re-title a conversation from later turns or from full transcripts.
• —Not a summariser. It produces labels, not descriptions.
• —Inherits Qwen3-0.6B's knowledge cutoff and general limitations.

Usage

MLX

from mlx_lm import load, generate
import re

model, tok = load("VertexAGI/prism-caption-1-5-micro")

SYSTEM = ("You name chat conversations. Given the user's first message, reply with ONLY a "
          "short, specific chat title (3-6 words, title case, no quotes, no punctuation at "
          "the end, no preamble). Nothing else -- just the title.")

msgs = [{"role": "system", "content": SYSTEM},
        {"role": "user", "content": "Hey, why does my car make a grinding noise when braking?"}]
text = tok.apply_chat_template(msgs, add_generation_prompt=True, tokenize=False,
                               enable_thinking=False)
out = generate(model, tok, prompt=text, max_tokens=24, verbose=False)
print(re.sub(r"<think>.*?</think>", "", out, flags=re.S).strip())
# -> Car Braking Grinding Noise

Qwen3 emits an empty <think></think> block even in non-thinking mode — strip it before using the output.

GGUF (llama.cpp)

Use llama.cpp's own chat template rather than hand-building the prompt; a hand-rolled template makes the model emit a thinking preamble and degrades the title.

llama-cli -m prism_caption_1_5_Q4_K_M.gguf \
  -sys "You name chat conversations. Given the user's first message, reply with ONLY a short, specific chat title (3-6 words, title case, no quotes, no punctuation at the end, no preamble). Nothing else -- just the title." \
  -p "Hey, why does my car make a grinding noise when braking?" \
  -n 20 --temp 0 -st --chat-template-kwargs '{"enable_thinking":false}'
# -> Car Braking Noise

Formats

License

Apache 2.0, inherited from Qwen3.