allenai / allenai/RL4LMs

Persistent Variance in IMDB

Abierto
#37 1 comentario 0 reacciones 0 asignados Ver en GitHub
Lenguaje dominante
Python
Estrellas
2.4k
Forks
201
Métricas de merge de PR
Sin PR fusionados en 30 d

Descripción

In running experiments on IMDB, I found that there was a very high variance in validation and test set results and I don't fully understand it, so I'm looking for some advice.

Here, I've run PPO for 10 seeds using default hyperparameters

![image](https://user-images.githubusercontent.com/3391297/216439583-10cfc758-555b-49f5-8471-a2d67fa0bd6c.png)

First of all, its clear that
1. there is clearly a large variance in performance at epoch 0, which could be explained by randomness in the eval sampling during decoding
2. there is a large variance in performance at epoch 50, which could be explained by randomness in RL

But together, we see runs that perform best at epoch 0 generally perform best on perplexity at epoch 50, which I can't explain. Here's the top 5 and bottom 5 based on initial perplexity scores, plotted against each other

![image](https://user-images.githubusercontent.com/3391297/216446969-33f79486-7853-40ef-8583-001b91df17a1.png)

Given that all models should be initialized to the pretrained model, there should be no randomness in initialization. So I'm confused as to how this is possible. Getting a lucky random seed for the initial validation should not affect the random seed for RL training, so why does the model that performs best at epoch 0 generally perform best at epoch 50?

Finally, I think the variance in results is high enough that I would recommend using 10 seeds for RL4LMs experiments

Guía de contribución

No hay ninguna guía de contribución indexada para este repositorio

Evaluación

Este issue todavía no se ha evaluado.

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.