---
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},
}
```