wiener_lpdf full model returns -inf for valid-support inputs in variability branches
Nobody has claimed this yet.
- Dominant language
- C++
- Stars
- 839
- Forks
- 220
- Avg merge
- 2d 4h
- Merged PRs (30d)
- 14
Description
Summary
A large combinatorial support sweep of the full Wiener wiener_lpdf(y, a, t0, w, v, sv, sw, st0, eps) implementation found many valid-support inputs for which the primal double-valued log density returns -inf.
This appears to be a primal numerical issue. I have discovered the issue while discussing #3322 with @Franzi2114 and @SteveBronder.
Grid Result
The sweep covered the full configured grid:
- valid support cases processed: 42,611,184 / 42,611,184
- shards completed: 512 / 512
- non-finite primal log density cases: 10,011,627
- exceptions: 0
- non-finite autodiff logp after excluding non-finite primal cases: 0
- gradient-only non-finite cases: 0
- finite-difference checks: 0 in this run
The failures were evenly distributed over supplied precision:
eps = 1e-4: 3,337,209eps = 1e-6: 3,337,209eps = 1e-8: 3,337,209
So this does not look like the supplied precision is the problem.
Failure Locations
Failures occur in numerically difficult but valid parts of the support. They are
frequent for, among others:
y - t0 = 1e-6,1e-4,10,50a = 0.1,8v = +/-30,+/-10
A small example from the sweep:
y=1e-6, a=0.1, t0=0, w=0.0001, v=-10, sv=0.1, sw=1e-8, st0=0.001, eps=1e-8
returns -inf from the primal double-valued lpdf.
Suspected Code Path
The issue appears to be in the full Wiener variability path in:
stan/math/prim/prob/wiener_full_lpdf.hpp
The full model computes a natural-scale density through integration and then takes the log:
density = internal::wiener7_integrate(...);
log_density += log(density);
The integrand is based on:
internal::wiener5_density<GradientCalc::ON>(...)
where GradientCalc::ON == true, so this uses the natural-scale branch:
return NaturalScale ? exp(log_density) : log_density;
in stan/math/prim/prob/wiener5_lpdf.hpp.
This means very small but finite log-density values can be converted to 0.0 before
or during integration, after which the public lpdf returns log(0) = -inf.
The 5-parameter path is less exposed because it accumulates the log density directly.
Expected Behavior
For valid-support inputs with positive mathematical density, the log density should
be finite. If a region is numerically too difficult, the implementation should avoid
silent natural-scale underflow to -inf or use a rescaled/log-domain integration
strategy.
Artifacts
I can attach the generated report, test code, and analysis from the grid run. The raw
failure CSVs are large, but can be sampled if useful.
Plan
I plan to submit a fix, possibly with this strategy:
- Add scaled fixed-quadrature fallback for
wiener_fullwhen adaptive returns zero/non-finite density. - Make it opt-in or policy-controlled? - this is a design question that needs to be answered, since the region is quite extreme and might never come up in empirical data, but users should be notified about this behavior/policy if left unchanged.
- Benchmark ordinary cases to confirm no measurable slowdown when fallback is not triggered.
- Add a regression test for the failing finite-support case.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reproducing the supplied finite-support case in stan/math/prim/prob/wiener_full_lpdf.hpp and trace its integration into wiener5_lpdf.hpp. Compare the natural-scale path with the direct log-density path, then add the proposed regression test. Done means the valid case returns a finite primal log density without regressing ordinary cases.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp
- Domain
- backend
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Quiet
- Clarity
- Mostly clear
- Newbie friendliness
- 45/100