Update roadmap and TODO for v1.0.2 with future focus

- Remove completed items tracking (now in CHANGELOG.md)
- Reorganize roadmap into Near-Term, Medium-Term, Long-Term sections
- Add new research-backed future enhancements:
  - Continuous Treatment DiD (Callaway, Goodman-Bacon, Sant'Anna 2024)
  - de Chaisemartin-D'Haultfœuille (reversible treatments)
  - Nonlinear DiD (Wooldridge 2023)
  - Matrix Completion Methods (Athey et al.)
  - Causal Forests for HTE in DiD settings
- Clean up TODO.md by removing completed items
- Focus on actionable technical debt and improvements

Author: claude

ROADMAP.md

@@ -2,190 +2,192 @@
 
 This document outlines the feature roadmap for diff-diff, prioritized by practitioner value and academic credibility.
 
-## What Makes a Credible 1.0?
+For past changes and release history, see [CHANGELOG.md](CHANGELOG.md).
 
-A production-ready DiD library needs:
-
-1. ✅ **Core estimators** - Basic DiD, TWFE, MultiPeriod, Staggered (Callaway-Sant'Anna), Synthetic DiD
-2. ✅ **Valid inference** - Robust SEs, cluster SEs, wild bootstrap for few clusters
-3. ✅ **Assumption diagnostics** - Parallel trends tests, placebo tests
-4. ✅ **Sensitivity analysis** - What if parallel trends is violated? (Rambachan-Roth)
-5. ✅ **Conditional parallel trends** - Covariate adjustment for staggered DiD
-6. ✅ **Documentation** - API reference site for discoverability
+---
 
-**All 1.0 blockers are complete.** diff-diff has feature parity with R's `did` + `HonestDiD` ecosystem for core DiD analysis.
+## Current Status (v1.0.2)
 
----
+diff-diff is a **production-ready** DiD library with feature parity with R's `did` + `HonestDiD` ecosystem for core DiD analysis:
 
-## Status Overview
-
-| Feature | Status | Priority | Why It Matters |
-|---------|--------|----------|----------------|
-| Honest DiD (Rambachan-Roth) | ✅ Done | — | Reviewers expect sensitivity analysis |
-| CallawaySantAnna Covariates | ✅ Done | — | Conditional PT often required in practice |
-| API Documentation Site | ✅ Done | — | Credibility and discoverability |
-| Goodman-Bacon Decomposition | ✅ Done | — | Explains when TWFE fails |
-| Power Analysis | ✅ Done | — | Study design tool |
-| CallawaySantAnna Bootstrap | ✅ Done | — | Valid inference with few clusters |
-| Sun-Abraham Estimator | Not Started | Post-1.0 | Alternative to CS, some prefer it |
-| Gardner's did2s | Not Started | Post-1.0 | Two-stage approach, available in pyfixest |
-| Local Projections DiD | Not Started | Post-1.0 | Dynamic effects (Dube et al. 2023) |
-| Borusyak-Jaravel-Spiess | Not Started | Post-1.0 | More efficient under homogeneous effects |
-| Double/Debiased ML | Not Started | Post-1.0 | High-dimensional covariates |
+- **Core estimators**: Basic DiD, TWFE, MultiPeriod, Callaway-Sant'Anna, Synthetic DiD
+- **Valid inference**: Robust SEs, cluster SEs, wild bootstrap, multiplier bootstrap
+- **Assumption diagnostics**: Parallel trends tests, placebo tests, Goodman-Bacon decomposition
+- **Sensitivity analysis**: Honest DiD (Rambachan-Roth)
+- **Study design**: Power analysis tools
 
 ---
 
-## 1.0 Target Features
+## Near-Term Enhancements (v1.1–v1.2)
 
-These would strengthen the 1.0 release but aren't strictly blocking.
+High-value additions building on our existing foundation.
 
-### ✅ Goodman-Bacon Decomposition (Done)
+### Sun-Abraham Estimator
 
-Helps users understand *why* TWFE can be biased with staggered adoption. Shows weights on "forbidden comparisons" (already-treated as controls). Essential diagnostic before deciding whether to use Callaway-Sant'Anna.
+Interaction-weighted estimator providing an alternative to Callaway-Sant'Anna. Many practitioners run both as a robustness check.
 
-- ✅ Decompose TWFE into 2x2 comparisons
-- ✅ Show weights by comparison type (clean vs. forbidden)
-- ✅ Visualization of decomposition (scatter and bar charts)
-- ✅ Integration with `TwoWayFixedEffects.decompose()` method
-- ✅ Automatic warning when TWFE detects staggered treatment timing
+- Event-study coefficients via saturated regression with cohort-time interactions
+- Different weighting scheme than CS; can give different results under heterogeneous effects
+- Useful robustness check when CS and SA agree
 
-**Reference**: Goodman-Bacon (2021). *Journal of Econometrics*.
+**Reference**: Sun & Abraham (2021). *Journal of Econometrics*.
 
-### ✅ Power Analysis Tools (Done)
+### Borusyak-Jaravel-Spiess Imputation Estimator
 
-Practitioners need to know "how many units/periods do I need to detect an effect of size X?" Now available in diff-diff.
+More efficient than Callaway-Sant'Anna when treatment effects are homogeneous across groups/time. Uses imputation rather than aggregation.
 
-- ✅ Minimum detectable effect given sample size
-- ✅ Required sample size for target power
-- ✅ Simulation-based power for any estimator (including staggered designs)
-- ✅ Visualization of power curves
-- ✅ Panel data considerations (ICC, multiple periods)
+- Imputes untreated potential outcomes using pre-treatment data
+- More efficient under homogeneous effects assumption
+- Can handle unbalanced panels more naturally
 
-**References**: Bloom (1995); Burlig, Preonas, & Woerman (2020).
+**Reference**: Borusyak, Jaravel, and Spiess (2024). *Review of Economic Studies*.
 
-### ✅ CallawaySantAnna Bootstrap Inference (Done)
+### Gardner's Two-Stage DiD (did2s)
 
-With few clusters or groups, analytical SEs may be unreliable. Multiplier bootstrap provides valid inference following the R `did` package approach.
+Two-stage approach gaining traction in applied work. First residualizes outcomes, then estimates effects.
 
-- ✅ Multiplier bootstrap at unit level with influence function perturbation
-- ✅ Aggregate bootstrap samples for overall ATT, event study, and group effects
-- ✅ Rademacher, Mammen, and Webb weight distributions
-- ✅ Percentile confidence intervals and bootstrap p-values
+- Stage 1: Estimate unit and time FEs using only untreated observations
+- Stage 2: Regress residualized outcomes on treatment indicators
+- Clean separation of identification and estimation
 
-**Reference**: Callaway & Sant'Anna (2021). *Journal of Econometrics*.
+**Reference**: Gardner (2022). *Working Paper*.
 
 ### Enhanced Visualization
 
 - Synthetic control weight visualization (bar chart of unit weights)
-- ✅ Bacon decomposition visualization (scatter and bar charts)
-- Treatment adoption "staircase" plot
+- Treatment adoption "staircase" plot for staggered designs
+- Interactive plots with plotly backend option
 
 ---
 
-## Post-1.0 Features
+## Medium-Term Enhancements (v1.3+)
 
-These are valuable but can wait for future versions.
+Extending diff-diff to handle more complex settings.
 
-### Sun-Abraham Estimator
+### Continuous Treatment DiD
 
-Alternative to Callaway-Sant'Anna using interaction-weighted approach. Some practitioners prefer it; provides a robustness check.
+Many treatments have dose/intensity rather than binary on/off. Active research area with recent breakthroughs.
 
-**Reference**: Sun & Abraham (2021). *Journal of Econometrics*.
+- Treatment effect on treated (ATT) parameters under generalized parallel trends
+- Dose-response curves and marginal effects
+- Handle settings where "dose" varies across units and time
+- Event studies with continuous treatments
 
-### Gardner's Two-Stage DiD (did2s)
+**References**:
+- [Callaway, Goodman-Bacon & Sant'Anna (2024)](https://arxiv.org/abs/2107.02637). *NBER Working Paper*.
+- [de Chaisemartin, D'Haultfœuille & Vazquez-Bare (2024)](https://arxiv.org/abs/2402.05432). *AEA Papers and Proceedings*.
+
+### de Chaisemartin-D'Haultfœuille Estimator
+
+Handles treatment that switches on and off (reversible treatments), unlike most other methods.
 
-Two-stage approach to staggered DiD that first residualizes outcomes using untreated observations, then estimates treatment effects. Available in pyfixest (Python) and did2s (R).
+- Allows units to move into and out of treatment
+- Time-varying, heterogeneous treatment effects
+- Comparison with never-switchers or flexible control groups
+- Different assumptions than CS/SA—useful for different settings
 
-**Reference**: Gardner (2022). *Two-stage differences in differences*.
+**Reference**: [de Chaisemartin & D'Haultfœuille (2020, 2024)](https://papers.ssrn.com/sol3/papers.cfm?abstract_id=3980758). *American Economic Review*.
 
 ### Local Projections DiD
 
-Implements local projections for dynamic treatment effects. Flexible approach that doesn't require specifying the full dynamic structure. Gaining traction in applied work.
+Implements local projections for dynamic treatment effects. Doesn't require specifying full dynamic structure.
+
+- Flexible impulse response estimation
+- Robust to misspecification of dynamics
+- Natural handling of anticipation effects
+- Growing use in macroeconomics and policy evaluation
 
 **Reference**: Dube, Girardi, Jordà, and Taylor (2023).
 
-### Borusyak-Jaravel-Spiess Imputation Estimator
+### Nonlinear DiD
 
-More efficient than Callaway-Sant'Anna when parallel trends holds across all periods. Uses imputation approach.
+For outcomes where linear models are inappropriate (binary, count, bounded).
 
-**Reference**: Borusyak, Jaravel, and Spiess (2024).
+- Logit/probit DiD for binary outcomes
+- Poisson DiD for count outcomes
+- Flexible strategies for staggered designs with nonlinear models
+- Proper handling of incidence rate ratios and odds ratios
 
-### Double/Debiased ML for DiD
+**Reference**: [Wooldridge (2023)](https://academic.oup.com/ectj/article/26/3/C31/7250479). *The Econometrics Journal*.
 
-For high-dimensional settings with many covariates. Uses ML for nuisance parameter estimation with cross-fitting.
+---
 
-**Reference**: Chernozhukov et al. (2018), Chang (2020).
+## Long-Term Research Directions (v2.0+)
 
-### Alternative Inference Methods
+Frontier methods requiring more research investment.
 
-- Randomization inference for small samples
-- Bayesian DiD with prior on parallel trends
-- Conformal inference for prediction intervals
+### Matrix Completion Methods
 
----
+Unified framework encompassing synthetic control and regression approaches. Moves seamlessly between cross-sectional and time-series patterns.
 
-## Release History
+- Nuclear norm regularization for low-rank structure
+- Handles missing data patterns common in panel settings
+- Bridges synthetic control (few units, many periods) and regression (many units, few periods)
+- Confidence intervals via debiasing
 
-### v0.9.0 (Current)
+**Reference**: [Athey et al. (2021)](https://arxiv.org/abs/1710.10251). *Journal of the American Statistical Association*.
 
-- ✅ Callaway-Sant'Anna multiplier bootstrap inference
-- ✅ Rademacher, Mammen, and Webb weight distributions
-- ✅ Bootstrap SEs, CIs, and p-values for all aggregations (overall ATT, event study, group effects)
-- ✅ `CSBootstrapResults` dataclass for bootstrap results
+### Causal Forests for DiD
 
-### v0.8.0
+Machine learning methods for discovering heterogeneous treatment effects in DiD settings.
 
-- ✅ Power analysis tools (`PowerAnalysis`, `simulate_power`)
-- ✅ MDE, sample size, and power calculations
-- ✅ Simulation-based power for any DiD estimator
-- ✅ Power curve visualization (`plot_power_curve`)
-- ✅ Panel data support with ICC adjustment
+- Estimate treatment effect heterogeneity across covariates
+- Data-driven subgroup discovery
+- Combine with DiD identification for observational data
+- Honest confidence intervals for discovered heterogeneity
 
-### v0.7.0
+**References**:
+- [Kattenberg, Scheer & Thiel (2023)](https://ideas.repec.org/p/cpb/discus/452.html). *CPB Discussion Paper*.
+- Athey & Wager (2019). *Annals of Statistics*.
 
-- ✅ Goodman-Bacon decomposition for TWFE diagnostics
-- ✅ `plot_bacon()` visualization (scatter and bar charts)
-- ✅ `TwoWayFixedEffects.decompose()` integration
-- ✅ Automatic staggered treatment warning in TWFE
+### Double/Debiased ML for DiD
+
+For high-dimensional settings with many potential confounders.
+
+- ML for nuisance parameter estimation (propensity, outcome models)
+- Cross-fitting for valid inference
+- Handles many covariates without overfitting concerns
+- Doubly-robust estimation with ML flexibility
 
-### v0.6.0
+**Reference**: Chernozhukov et al. (2018). *The Econometrics Journal*.
 
-- ✅ **All 1.0 Blockers Complete**
-- ✅ Honest DiD sensitivity analysis (Rambachan & Roth 2023)
-- ✅ CallawaySantAnna covariate adjustment (DR, IPW, Reg)
-- ✅ API documentation site with Sphinx
+### Alternative Inference Methods
 
-### v0.5.0
+- **Randomization inference**: Exact p-values for small samples
+- **Bayesian DiD**: Priors on parallel trends violations
+- **Conformal inference**: Prediction intervals with finite-sample guarantees
 
-- Wild cluster bootstrap (Rademacher, Webb, Mammen weights)
-- Placebo tests module
-- Tutorial notebooks
+---
 
-### v0.4.0
+## Infrastructure Improvements
 
-- Callaway-Sant'Anna estimator for staggered DiD
-- Event study and group effects visualization
-- Parallel trends testing utilities
+Ongoing maintenance and developer experience.
 
-### v0.3.0
+### Performance
 
-- Synthetic Difference-in-Differences
-- Multi-period DiD with event study
-- Data preparation utilities
+- JIT compilation for bootstrap loops (numba)
+- Parallel bootstrap iterations
+- Sparse matrix handling for large fixed effects
+- Memory-efficient estimation for large panels
 
-### v0.2.0
+### Code Quality
 
-- Two-Way Fixed Effects estimator
-- Fixed effects support (absorb parameter)
-- Cluster-robust standard errors
-- Formula interface
+- Extract shared within-transformation logic to utils
+- Consolidate linear regression helpers
+- Consider splitting `staggered.py` (1800+ lines)
 
-### v0.1.0
+### Documentation
 
-- Initial release with basic DiD estimator
+- Real-world data examples (beyond synthetic)
+- Performance benchmarks vs. R packages
+- Video tutorials and worked examples
 
 ---
 
 ## Contributing
 
-Interested in contributing? See the [GitHub repository](https://github.com/igerber/diff-diff) for open issues. Features marked "Not Started" are good candidates for contributions.
+Interested in contributing? Features in the "Near-Term" and "Medium-Term" sections are good candidates. See the [GitHub repository](https://github.com/igerber/diff-diff) for open issues.
+
+Key references for implementation:
+- [Roth et al. (2023)](https://www.sciencedirect.com/science/article/abs/pii/S0304407623001318). "What's Trending in Difference-in-Differences?" *Journal of Econometrics*.
+- [Baker et al. (2025)](https://arxiv.org/pdf/2503.13323). "Difference-in-Differences Designs: A Practitioner's Guide."