| --- |
| library_name: newsrex |
| tags: |
| - news-recommendation |
| - naml |
| - jax |
| - mind |
| datasets: |
| - mind |
| license: apache-2.0 |
| --- |
| |
| # NewsReX NAML — JAX — MIND-small |
|
|
| NAML news recommendation model trained on MIND-small using the |
| [NewsReX](https://github.com/igor17400/NewsReX) framework (JAX). |
|
|
| ## Test Results |
|
|
| | Seed | AUC | MRR | NDCG@5 | NDCG@10 | |
| |------|-----|-----|--------|---------| |
| | 123 | 0.6640 | 0.3134 | 0.3457 | 0.4103 | |
| | 42 * | 0.6656 | 0.3155 | 0.3496 | 0.4124 | |
| | 456 | 0.6622 | 0.3101 | 0.3415 | 0.4064 | |
| | **mean ± std** | **0.6639±0.0014** | **0.3130±0.0022** | **0.3456±0.0033** | **0.4097±0.0025** | |
|
|
| \* Best seed (weights at repo root) |
|
|
| ## Experiment Configuration |
|
|
| ```yaml |
| model: |
| name: naml |
| architecture: |
| news_encoder: |
| type: cnn_multi_view |
| views: |
| title: |
| type: cnn |
| filter_num: 400 |
| kernel_size: 3 |
| activation: relu |
| attention_query_dim: 200 |
| abstract: |
| type: cnn |
| filter_num: 400 |
| kernel_size: 3 |
| activation: relu |
| attention_query_dim: 200 |
| category: |
| type: embedding |
| embedding_dim: 100 |
| subcategory: |
| type: embedding |
| embedding_dim: 100 |
| view_attention_query_dim: 200 |
| user_encoder: |
| type: additive_attention |
| attention_query_dim: 200 |
| num_attention_heads: 10 |
| click_predictor: |
| type: dot_product |
| embedding: |
| size: 300 |
| trainable: true |
| dropout_rate: 0.2 |
| seed: 42 |
| inputs: |
| title: |
| max_length: 32 |
| abstract: |
| max_length: 50 |
| history: |
| max_length: 50 |
| impressions: |
| max_length: 5 |
| process_title: true |
| process_abstract: true |
| process_category: true |
| process_subcategory: true |
| process_user_id: false |
| training: |
| loss: |
| name: categorical_crossentropy |
| from_logits: true |
| reduction: sum_over_batch_size |
| label_smoothing: 0.0 |
| optimizer: adam |
| learning_rate: 0.0001 |
| batch_size: 256 |
| num_epochs: 20 |
| early_stopping: |
| patience: 5 |
| min_improvement: 0.01 |
| negative_sampling: |
| strategy: random |
| candidates: 4 |
| evaluation: |
| mode: fast |
| metrics: |
| - auc |
| - mrr |
| - ndcg@5 |
| - ndcg@10 |
| batch_size: 512 |
| ``` |
|
|
| ## Repository Structure |
|
|
| ``` |
| newsrex/NAML-JAX-MIND-small/ |
| ├── model.safetensors ← best seed (42) |
| ├── test_results.json |
| ├── training_run_summary.json |
| ├── seed_123/model.safetensors |
| ├── seed_42/model.safetensors |
| ├── seed_456/model.safetensors |
| └── README.md |
| ``` |
|
|
| ## Usage |
|
|
| ```bash |
| git clone https://github.com/igor17400/NewsReX.git |
| cd NewsReX && uv sync |
| |
| # Run evaluation with best seed weights |
| uv run python src/eval.py \ |
| experiment=mind/naml \ |
| framework=jax \ |
| weights=hf://newsrex/NAML-JAX-MIND-small/model.safetensors |
| |
| # Run evaluation with a specific seed |
| uv run python src/eval.py \ |
| experiment=mind/naml \ |
| framework=jax \ |
| weights=hf://newsrex/NAML-JAX-MIND-small/seed_42/model.safetensors |
| ``` |
|
|
| ## Citation |
|
|
| ```bibtex |
| @misc{newsrex2026, |
| title={NewsReX: An Open-Source Multi-Framework for Neural News Recommendation}, |
| author={Igor L. R. Azevedo and Toyotaro Suzumura and Yuichiro Yasui}, |
| year={2025}, |
| eprint={2508.21572}, |
| archivePrefix={arXiv}, |
| primaryClass={cs.IR}, |
| url={https://arxiv.org/abs/2508.21572}, |
| } |
| ``` |
|
|
