Antevisão · Latent Rehearsal
19

A inversão
que não se repetiu

A apostila 15 registrou que, com dados suficientes, o modelo de mundo deixava de custar ao gerador e passava a render 3,38%. Era uma semente só. Com mais sementes, e depois com uma medida de validação que parou de oscilar, o custo aparece em todas as rodadas — e cresce com o tamanho do modelo.

Apostila 19 Pré-requisito: 15 (o que custa e o que entrega) e 13 §12.3 (sementes)

1 O que estava escrito

A apostila 15 treinou o mini no corpus inteiro, com o orçamento de Chinchilla (1.961 passos, 64 milhões de tokens), de duas formas: só o gerador, e o gerador com o modelo de mundo hierárquico de alvo EMA — a receita que tinha ido melhor na apostila 13. O gerador sozinho chegou a 3,7503 de perda de validação; com o modelo de mundo, a 3,6236. Isso é 3,38% melhor, e a apostila leu o resultado como uma inversão: no treino curto da coleção o mapa custava cerca de 1%, e com dados suficientes passaria a render.

A leitura tinha uma explicação razoável — com poucos dados, prever o próximo estado compete com o gerador pela mesma capacidade; com muitos, funciona como regularizador — e combinava com o que se espera de uma arquitetura preditiva. Mas vinha de uma rodada de cada lado, com uma semente só, num regime em que o desvio entre sementes nunca tinha sido medido. A apostila 13 §12.3 mediu esse desvio no duelo: cerca de 0,84% da perda. Na escada, ninguém tinha medido.

2 Duas rodadas

Em 17/09, numa RTX 5090, o mesmo par foi treinado com as sementes 2 e 3, com os mesmos passos, o mesmo lote e o mesmo corpus. Antes disso, a rodada repetiu o controle do duelo, que deu 5,2577 de validação, o mesmo número da placa de 11/09 até a quarta casa: a máquina não é a fonte da diferença. Essa rodada trouxe também as medidas das apostilas 13 e 16.

O resultado dela apontou um problema na própria régua (§4), e por isso houve uma segunda rodada no mesmo dia, com a validação medida de outro jeito: seis rodadas do mini e quatro do pequeno, todas com --eval-seed, a opção nova que faz toda avaliação ler os mesmos lotes.

3 Três sementes, com a validação sorteada

Perda de validação do mini na escada (64 M tokens), por semente, no protocolo original: a cada avaliação o treino sorteia 20 lotes novos do conjunto de validação. "Melhor" é o menor valor entre as 12 avaliações, que é o que o relatório usa. Diferença = quanto o modelo com o mapa fica acima (+) ou abaixo (−) do gerador sozinho.
sementesó gerador+ mapadiferença
1337 (a da apostila 15)3,75033,6236−3,38%
23,57623,6401+1,79%
33,60913,7974+5,22%
média ± desvio3,6452 ± 0,09253,6870 ± 0,0959+1,15%

Os dados não confirmaram a inversão. Nas duas sementes novas o mapa custa, e a média das três fica em +1,15% — o mesmo custo que a apostila 13 tinha medido no treino curto. A semente 1337 é a única em que o modelo com o mapa fica abaixo, e é também a semente em que o gerador sozinho teve o pior resultado das três (3,7503).

Só que o desvio aqui é grande demais para afirmar qualquer coisa: 0,0925 no gerador sozinho, 2,5% da perda, contra os 0,84% do duelo. A diferença entre as médias, 0,042, é menos da metade desse desvio. Com esta régua, nem a inversão nem o custo se sustentam.

4 Por que uma semente enganou

O desvio grande não vem só da inicialização. Ele começa dentro de cada rodada. A cada avaliação, o treino sorteia 20 lotes novos do conjunto de validação e mede a perda neles; duas avaliações seguidas do mesmo modelo quase não mudam os pesos, mas mudam a amostra.

As seis últimas avaliações de três rodadas do mini na escada, com a validação sorteada. O mínimo de cada linha está em negrito.
rodadapasso 114713111475163918031960
só gerador, semente 23,9643,8193,7353,5763,6483,719
só gerador, semente 13373,9193,8783,7503,7643,7993,759
+ mapa, semente 24,0203,7733,6873,8213,6403,780

Na semente 2, o gerador sozinho marca 3,576 no passo 1639 e 3,719 no fim do treino. São 0,14 de diferença entre duas medidas do mesmo modelo quase pronto — mais que os 0,127 que separavam as colunas da semente 1337 e sustentavam a inversão. O "melhor checkpoint" é o menor de 12 números desse tipo: ele escolhe, em cada rodada, a avaliação que caiu na amostra mais fácil. Comparar dois mínimos de medidas ruidosas compara, em boa parte, quem teve mais sorte no sorteio.

O conserto é pequeno e aditivo: model/train.py --eval-seed <n> faz toda avaliação ler os mesmos lotes de validação. Sem a opção, o comportamento continua o de antes, para não quebrar a comparação com o que já está registrado.

5 A mesma pergunta, com a régua limpa

Com os lotes fixos, o par inteiro foi treinado de novo: três sementes do mini e duas do pequeno, dos dois lados. Os números desta seção não se comparam com os da §3 — são amostras de validação diferentes —, mas se comparam entre si, que é o que a pergunta pede.

Perda de validação com --eval-seed, corpus completo. Com lotes fixos a curva desce sem picos, e o melhor checkpoint é sempre o último. mini: 1.961 passos, 64 M tokens. pequeno: 6.483 passos, 212 M tokens.
modelosementesó gerador+ mapadiferença
mini13373,67293,6841+0,30%
23,62373,7141+2,50%
33,61363,7051+2,53%
média ± desvio3,6367 ± 0,03183,7011 ± 0,0154+1,77%
pequeno13372,94783,0978+5,09%
22,94753,0875+4,75%
média ± desvio2,9477 ± 0,00023,0927 ± 0,0073+4,92%

Três coisas mudam de status com esta tabela.

As curvas de validação da semente 1337 nos dois tamanhos, com lotes fixos. Cada coluna é uma avaliação.
rodada123456789101112
mini, só gerador5,7245,0694,7214,4794,2594,0733,9363,8353,7663,7203,6923,673
mini, + mapa5,7785,1174,7624,4954,2274,0153,8883,8073,7563,7233,6993,684
pequeno, só gerador4,3423,5983,3933,2833,1963,1373,0843,0373,0042,9752,9602,948
pequeno, + mapa4,4853,6343,4163,3103,2413,1973,1593,1373,1173,1053,1003,098

As curvas mostram onde o custo nasce. No mini, o modelo com o mapa vai melhor no meio do treino (3,888 contra 3,936 na sétima avaliação) e perde só no fim, quando o gerador sozinho continua descendo e ele desacelera. No pequeno, o mesmo acontece mais cedo e mais forte: da sexta avaliação em diante a curva com o mapa achata, e a distância entre as duas cresce até o fim. Quanto maior o modelo, mais cedo a perda do mapa parece competir com a do gerador.

A régua que muda

A regra da apostila 13 §12.3 — diferença abaixo de cerca de 0,9% da perda não é resultado — foi medida no duelo, com a validação sorteada, e continua valendo para o que está registrado com aquela régua. Com lotes de validação fixos, o desvio entre sementes na escada é o mesmo 0,87% no mini e quase zero no pequeno, e diferenças pequenas voltam a ser legíveis. Daqui para a frente, rodada de escada sai com --eval-seed.

6 O que muda na coleção

Onde a leitura de uma semente estava, e como ficou.
ondeantesagora
15 §5"com orçamento de verdade, o modelo de mundo para de custar e passa a render": −3,38%tabela com as duas réguas: o mapa custa +1,77% no mini e +4,92% no pequeno; a leitura de inversão foi retirada
17, tabela das medidas"3,38% melhor (PPL 37,5 contra 42,5)"o custo do mapa ao gerador, com validação fixa, nos dois tamanhos
model/train.pylotes de validação sorteados a cada avaliaçãoopção --eval-seed: lotes fixos; sem ela, o comportamento de antes

As frases da coleção sobre o custo do modelo de mundo ao gerador, medidas com 5,7 milhões de tokens, continuam valendo com 64 e com 212 milhões. O que não continua é a ideia de que mais dados, sozinhos, transformam esse custo em ganho: com esta receita, o custo persiste e aumenta com o tamanho. O que o mapa entrega em troca está medido em outras apostilas — o leme que governa a geração (12), a violação de expectativa e o guardrail (13) — e nada disso aparece na perda de validação.

7 Limites desta medida

Perguntas em aberto
Referências

As fontes das ideias e do método usados aqui. Os números foram medidos no projeto.

  1. Assran, M., Duval, Q., Misra, I., Bojanowski, P., Vincent, P., Rabbat, M., LeCun, Y. e Ballas, N. (2023). Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture. CVPR 2023 (I-JEPA, alvo de professor EMA). arXiv:2301.08243
  2. Grill, J.-B. et al. (2020). Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning. NeurIPS 2020 (BYOL, alvo por média móvel exponencial). arXiv:2006.07733
  3. Bouthillier, X. et al. (2021). Accounting for Variance in Machine Learning Benchmarks. MLSys 2021 (fontes de variância entre rodadas e quantas repetições uma comparação pede). arXiv:2103.03098
  4. Dodge, J., Ilharco, G., Schwartz, R., Farhadi, A., Hajishirzi, H. e Smith, N. (2020). Fine-Tuning Pretrained Language Models: Weight Initializations, Data Orders, and Early Stopping. (efeito da semente e da escolha do checkpoint). arXiv:2002.06305
  5. Apostila 15 desta coleção, O que custa e o que entrega, §5.

Treinos numa RTX 5090 em 17/09/2026 (cloud/round_1709.sh e cloud/round_1709b.sh). Dados em out/gpu_1709/, model/runs/*-escala e model/runs/*-escalafix.