Add model_card.md
Browse files- model_card.md +89 -0
model_card.md
ADDED
|
@@ -0,0 +1,89 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# UGTC Model Card
|
| 2 |
+
|
| 3 |
+
## Model Description
|
| 4 |
+
|
| 5 |
+
**Name:** UGTC (Uncertainty-Gated Temporal Credit)
|
| 6 |
+
**Type:** Reinforcement Learning Algorithm Component (Advantage Estimator)
|
| 7 |
+
**Author:** Yağız Ekrem Dalar
|
| 8 |
+
**License:** MIT
|
| 9 |
+
**Paper:** https://doi.org/10.5281/zenodo.19715116
|
| 10 |
+
|
| 11 |
+
### What is UGTC?
|
| 12 |
+
|
| 13 |
+
UGTC is a **plug-in advantage estimator** for actor-critic reinforcement learning algorithms.
|
| 14 |
+
It is not a standalone model but a module that replaces the standard advantage computation
|
| 15 |
+
in any actor-critic algorithm (PPO, TD3, SAC, DDPG, etc.).
|
| 16 |
+
|
| 17 |
+
### How it Works
|
| 18 |
+
|
| 19 |
+
UGTC maintains two value critics with different GAE λ values:
|
| 20 |
+
|
| 21 |
+
1. **Fast critic** (single MLP, λ=0.80): Low variance, higher bias
|
| 22 |
+
2. **Slow ensemble** (M=3 MLPs, λ=0.99): Lower bias, higher variance
|
| 23 |
+
|
| 24 |
+
A sigmoid gate `u(s) = σ(-β·(σ̂(s) - 1))` blends their advantage estimates:
|
| 25 |
+
|
| 26 |
+
```
|
| 27 |
+
A^UGTC_t = u(sₜ)·A^slow_t + (1-u(sₜ))·A^fast_t
|
| 28 |
+
```
|
| 29 |
+
|
| 30 |
+
where `σ̂(s)` is the EMA-normalized ensemble disagreement.
|
| 31 |
+
|
| 32 |
+
## Intended Use
|
| 33 |
+
|
| 34 |
+
- Research on advantage estimation in deep RL
|
| 35 |
+
- Integration into existing actor-critic implementations
|
| 36 |
+
- Educational exploration of uncertainty-based credit assignment
|
| 37 |
+
|
| 38 |
+
## Fixed Hyperparameters
|
| 39 |
+
|
| 40 |
+
| Parameter | Value | Description |
|
| 41 |
+
|-----------|-------|-------------|
|
| 42 |
+
| λ_fast | 0.80 | Fast critic GAE lambda |
|
| 43 |
+
| λ_slow | 0.99 | Slow ensemble GAE lambda |
|
| 44 |
+
| M | 3 | Ensemble size |
|
| 45 |
+
| β | 5.0 | Gate temperature |
|
| 46 |
+
| EMA momentum | 0.99 | Running uncertainty normalization |
|
| 47 |
+
|
| 48 |
+
These are fixed across **all benchmarks** — no per-task tuning.
|
| 49 |
+
|
| 50 |
+
## Training and Evaluation
|
| 51 |
+
|
| 52 |
+
UGTC has been integrated with PPO, TD3, and SAC and evaluated on:
|
| 53 |
+
- Procgen 16-game benchmark (hard mode, 25M steps)
|
| 54 |
+
- MetaWorld ML45 (multitask)
|
| 55 |
+
- DeepMind Control Suite (DMC)
|
| 56 |
+
- ManiSkill (manipulation multitask)
|
| 57 |
+
- MuJoCo: Hopper-v4, Ant-v5
|
| 58 |
+
- Image-based: CarRacing-v3
|
| 59 |
+
|
| 60 |
+
For specific results, see the paper: https://doi.org/10.5281/zenodo.19715116
|
| 61 |
+
|
| 62 |
+
## Limitations and Assumptions
|
| 63 |
+
|
| 64 |
+
- UGTC-DDPG is an **implementation extension** — not directly evaluated in the paper
|
| 65 |
+
- Performance gains depend on the specific environment and algorithm backbone
|
| 66 |
+
- The UGTC critics introduce a small parameter overhead (~3 MLP value heads)
|
| 67 |
+
- EMA normalization requires a warmup period for accurate σ_EMA estimation
|
| 68 |
+
- Performance may vary from paper results due to hardware, library versions, and random seed
|
| 69 |
+
|
| 70 |
+
## Ethical Considerations
|
| 71 |
+
|
| 72 |
+
UGTC is a research algorithm for reinforcement learning. It does not involve:
|
| 73 |
+
- Personal data
|
| 74 |
+
- Decision-making in high-stakes autonomous systems
|
| 75 |
+
- Generative content
|
| 76 |
+
|
| 77 |
+
## Publication
|
| 78 |
+
|
| 79 |
+
> **UGTC: Uncertainty-Gated Temporal Credit**
|
| 80 |
+
> Yağız Ekrem Dalar
|
| 81 |
+
> *Accepted — Ulysseus Young Explorers in Science (UYES) Journal*
|
| 82 |
+
> Preprint DOI: [10.5281/zenodo.19715116](https://doi.org/10.5281/zenodo.19715116)
|
| 83 |
+
> Journal DOI: Forthcoming
|
| 84 |
+
|
| 85 |
+
## Links
|
| 86 |
+
|
| 87 |
+
- GitHub: https://github.com/ethosoftai/ugtc
|
| 88 |
+
- Documentation: https://ethosoftai.github.io/ugtc
|
| 89 |
+
- Demo: https://huggingface.co/spaces/Ethosoft/ugtc
|