Validate strategy-memory flags under external sampling

OpenSpiel's Deep CFR records strategy samples at opponent nodes during
the traverser's tree walk; storing on traverser nodes under external
sampling drops the ρ_p reach factor and biases the average-policy
estimate. Reject that config combination at load time and add a research
note deriving why outcome sampling is unaffected while external is not.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
2026-05-07 20:56:57 +09:00
co-authored by Claude Opus 4.7
parent edad3b47da
commit 6c9babe769
2 changed files with 179 additions and 1 deletions
+162
View File
@@ -0,0 +1,162 @@
# Strategy Memory Recording Location: Traverser vs. Opponent Nodes
**Last verified:** 2026-05-07, commit `edad3b4`
Source: prompted by static-review feedback claiming our default
`store_strategy_on_traverser_nodes: true` is the reverse of the OpenSpiel
Deep CFR convention.
## Question
Our default config records strategy samples at the *traverser*'s own info
states during a traversal. OpenSpiel's reference Deep CFR records them at
the *opponent*'s info states. Does this difference bias the average-policy
estimate, and does the answer depend on `sampling_mode`?
Short answer: **for the current outcome-sampling default it is empirically
fine, but if we ever flip `sampling_mode: external` we must also flip these
two flags or the strategy memory will diverge from the OpenSpiel
convention.** The bias direction differs between sampling modes; the safe
rule is "use OpenSpiel's convention (opp nodes, opp info state) for
external; either is acceptable for outcome".
## Code reference
Our recording site,
`src/coolrl_lost_cities/games/classic/deep_cfr/traversal.pyx`,
`_record_strategy` (line 877):
```cython
cdef void _record_strategy(self, info_state, legal, policy, player,
traverser, iteration, depth, stats):
if player == traverser:
if not self.store_strategy_on_traverser_nodes:
return
elif not self.store_strategy_on_opponent_nodes:
return
...
self.strategy_samples.append(TrainingSample(
info_state=info_state, target=target, legal_mask=legal_mask,
iteration=iteration, player=player,
))
```
Called unconditionally at every decision node (line 323) before the
sampling-mode branch. The `info_state` is computed by `_policy(state,
player, ...)` for the *current acting player*, which is the right thing in
both conventions.
`configs/deep_cfr/default.yaml` sets:
```yaml
store_strategy_on_traverser_nodes: true
store_strategy_on_opponent_nodes: false
```
OpenSpiel reference
(`open_spiel/python/pytorch/deep_cfr.py::_traverse_game_tree`):
```python
elif state.current_player() == player:
# ... compute regrets, append to ADVANTAGE buffer
self._append_to_advantage_buffer(player, data)
return cfv
else:
other_player = state.current_player()
_, strategy = self._sample_action_from_advantage(state, other_player)
# ...
data = StrategyMemory(
np.array(state.information_state_tensor(other_player), ...),
np.array(self._iteration, ...),
np.array(strategy, dtype=np.float32),
)
self._append_to_stategy_buffer(data)
```
OpenSpiel: strategy samples are appended **only at opponent nodes** during
traverser=p's external-sampling tree walk, using the opponent's info state.
## Mechanism: why the convention differs by sampling mode
Average policy theory: π̄_p(I) = Σ_t ρ_p^t(I) · σ_p^t(I) / Σ_t ρ_p^t(I),
where ρ_p^t(I) is player p's contribution to the reach probability of I at
iteration t. The strategy network minimizes (CE/MSE) loss against samples
(I, σ_p^t(I)); the *empirical sample distribution* should be ∝ ρ_p(I) for
the regression to converge to π̄_p.
### External sampling (OpenSpiel default)
During p's traversal:
- At p-nodes, p enumerates all legal actions — so p's branching contributes
no probability factor. Visit count at p-info-state I scales as ρ_opp(I).
- At opp-nodes, opp samples one action ~ σ_opp. Visit count at opp-info-state
I scales as ρ_p(I) · ρ_opp(I).
To get π̄_opp samples weighted by ρ_opp, OpenSpiel records at opp-nodes
during p's traversal. The ρ_p factor on top is the iteration-mixing factor
(opp's strategy was fit while p was traversing) — over alternating-player
iterations this averages out as both players take turns being traverser.
If we instead recorded at p-nodes during p's traversal in external mode,
the empirical density would be ∝ ρ_opp(I_p) — which **drops the ρ_p
factor we wanted**, and overweights deep p-side info states reachable
mostly through opp-uniform play. That is the bias the reviewer flagged.
### Outcome sampling (our default)
During p's traversal both players sample on-policy along a single
trajectory:
- p-node info state I: visit prob ∝ ρ_p(I) · ρ_opp(I)
- opp-node info state I: visit prob ∝ ρ_p(I) · ρ_opp(I)
Both conventions give the same sample density up to the mixing factor of
the *other* player's reach. Neither is exactly ρ_p(I) without an
importance-sampling correction. The two are equivalent in expectation
modulo variance.
In practice we use uniform-priority reservoir sampling on the strategy
buffer rather than reach-weighted regression, so the residual ρ_other
factor is absorbed into "training-data distribution we accept" — it is
not a correctness bug, it is the Deep CFR approximation choice for both
conventions.
## Audit: what our default does in each mode
| `sampling_mode` | `store_on_traverser` | `store_on_opp` | OpenSpiel-equivalent? | Bias risk |
|---|---|---|---|---|
| `outcome` (current default) | true | false | n/a — OpenSpiel doesn't ship outcome Deep CFR | low; symmetric |
| `external` (if user flips) | **true** | **false** | **NO — reversed** | **high — drops ρ_p factor on π̄ samples** |
| `external` + flip flags | false | true | yes | matches reference |
## Practical implication
- The current default (outcome + traverser-node strategy) is not a bug for
the algorithm we are actually running. Keep it.
- Add a config validation that warns/errors when
`sampling_mode: external` is paired with
`store_strategy_on_traverser_nodes: true` or
`store_strategy_on_opponent_nodes: false`. This is the easy guardrail
that catches the reviewer's scenario.
- When introducing an `open_spiel_like.yaml` preset for parity
experiments, set:
```yaml
traversal:
sampling_mode: external
store_strategy_on_traverser_nodes: false
store_strategy_on_opponent_nodes: true
```
- Do **not** change the outcome-mode default to match OpenSpiel's
external-mode convention — that would be a cargo-cult fix; the
underlying reach-distribution argument does not transfer.
## References
- Brown, Lerer, Gross, Sandholm. *Deep Counterfactual Regret
Minimization.* ICML 2019. (Algorithm 1 — strategy memory M_Π
collection.)
- OpenSpiel `python/pytorch/deep_cfr.py`,
`DeepCFRSolver._traverse_game_tree` (commit master @ 2026-05-07).
- `docs/research/outcome-sampling-target.md` — companion note on the
advantage-target side of outcome sampling.
@@ -7,7 +7,7 @@ from pathlib import Path
from typing import Any from typing import Any
import yaml import yaml
from pydantic import BaseModel, ConfigDict, Field, field_validator from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
from coolrl_lost_cities.games.classic.game import LostCitiesConfig from coolrl_lost_cities.games.classic.game import LostCitiesConfig
@@ -159,6 +159,22 @@ class TraversalConfig(StrictModel):
raise ValueError("must be 'local' or 'server'") raise ValueError("must be 'local' or 'server'")
return token return token
@model_validator(mode="after")
def _validate_external_strategy_memory_convention(self) -> TraversalConfig:
if self.sampling_mode == "external" and (
self.store_strategy_on_traverser_nodes or not self.store_strategy_on_opponent_nodes
):
raise ValueError(
"sampling_mode='external' requires "
"store_strategy_on_traverser_nodes=False and "
"store_strategy_on_opponent_nodes=True to match the OpenSpiel "
"Deep CFR convention. Storing on traverser nodes under "
"external sampling drops the ρ_p reach factor and biases the "
"average-policy estimate. See "
"docs/research/strategy-memory-location.md."
)
return self
def resolved_num_workers(self, batches: int | None = None) -> int: def resolved_num_workers(self, batches: int | None = None) -> int:
if isinstance(self.num_workers, str): if isinstance(self.num_workers, str):
token = self.num_workers.strip().lower() token = self.num_workers.strip().lower()