Rose Pro

Architecture graph

A Rose That Finally Found the 1100 Club

I trained a model, watched the numbers climb,
And told myself, "This one will be prime."
I widened the model, increased the scale,
Then watched ArithMark politely fail.

Eighty billion tokens went into the machine,
Surely this time my model would be clean.
The Benchmarks smiled, the scores came through,
And said, "Nice little model. Here's some work for you."

Every time I think I've finally learned,
Another benchmark shows where I was wrong.
Truly, I am blessed beyond compare;
I keep discovering mistakes I can afford to train.

Introduction

Rose Pro is the latest model in the Rose X1 line, with approximately 151.3 million parameters.

It keeps the same general Rose X1 architecture while increasing the model's capacity, using 24 layers and a 640-dimensional hidden representation. The model was trained on approximately 80 billion tokens.

And yes, I had expectations.

Quite reasonable ones, obviously.

I had a bigger model, more capacity, a mountain of tokens, and enough confidence to become suspicious.

Then the Benchmarks arrived.

It did not exactly hand me a trophy.

It did, however, hand me something much more useful:

another collection of things to learn.

Model Overview

The architecture itself is intentionally not explained in excessive detail here. Rose Pro remains part of the Rose X1 family, and the interesting part of this release is less about repeating the architecture diagram and more about seeing what the model actually does with its increased capacity.

The Chemistry of Benchmarks

This is probably my favorite part of the whole release.

The Benchmarks are strange little ecosystem.

You put a model into it with millions of parameters, billions of tokens, a carefully designed architecture, and a considerable amount of optimism.

Then the Benchmarks removes the optimism and gives you percentages.

And suddenly you discover that one model is great at something, another model is better somewhere else, and a benchmark you thought would behave nicely has decided to personally insult you.

That is the chemistry of Becnhmarks.

Rose Pro does not dominate every benchmark, and I don't think it needs to.

What matters more is the overall shape of the results and what those results teach me about the model.

Benchmark Results

The following results are from 0-shot evaluation.

The results show a mixed but useful picture. Rose Pro moves strongly on several tasks while some benchmarks barely move and a few go backwards.

Which is precisely why I don't want to turn this README into a victory speech.

The model is good at some things.

It is still learning how to behave itself at others.

ArithMark 3 & BananaMind

These two deserve their own little section.

ArithMark 3

Rose Pro scores 38.20% on ArithMark 3.

Rose-Medium scored 38.30%.

So I increased the model.

I increased the width.

And ArithMark looked at all of that and said:

38.20%.

Thank you.

Very informative.

But that is also exactly why I find these evaluations useful. Parameter count does not magically translate into proportional improvement on every task.

Apparently I have to actually understand what is happening.

How inconvenient.

BananaMind Bench 1.1

And then there is the number I had been waiting for.

Rose-Medium: 1098.7 Elo

Rose Pro: 1105 Elo

I finally crossed 1100 Elo.

Is 1100 a scientifically meaningful boundary?

No.

Did 1098.7 bother me enough that I wanted to cross it?

Absolutely.

So I am going to enjoy this completely arbitrary victory.

1105.

Beautiful.

What I Learned This Time

One thing has remained remarkably consistent throughout my model-building journey:

Every time I train a model, I learn something.

Sometimes it is about architecture.

Sometimes it is about optimization.

Sometimes it is about evaluation.

And sometimes it is about a mistake that I would very much like to pretend I never made.

But every model leaves me with something new to investigate.

That is probably the most valuable part of building these models.

I start with an idea.

I train it.

I evaluate it.

Then the results politely inform me that my original understanding was only partially correct.

And honestly?

I think I am blessed.

Not because every model performs exactly how I imagined.

Quite the opposite.

I am blessed because every training run gives me another opportunity to discover something I didn't know before.

Apparently this is my preferred method of education.

It is just considerably more expensive than reading a book.

Training Recipe: One More Thing to Change

The Rose models so far have been trained using cosine learning-rate scheduling.

For the next generation, I want to try something different.

The upcoming models will explore WSD learning-rate scheduling.

Not because cosine suddenly became illegal.

I simply have another question I want to answer.

And apparently the only way I know how to answer questions is to build another model.

What's Next?

The Rose X1 journey is not finished.

The next step will be the Rose 1.5 series, which will explore further changes to the training recipe and continue the work started with the current Rose models.

Hopefully, the next leaderboard visit will contain fewer surprises.

Knowing me, that statement probably guarantees the opposite.

Final Thoughts

Rose Pro is not exactly the model I imagined when I started building it.

Some results are encouraging.

Some are competitive.

Some are stubborn.

And some are sitting there quietly reminding me that adding more parameters is not the same thing as solving the problem.

But that is fine.

The point of every model is not to prove that I was right.

Sometimes the point is to find out where I was wrong.

Rose Pro gave me another set of answers.

It also gave me several new questions.

And, most importantly:

1105 Elo.

I'll take it.

For now.