feat(ml): integrate predictive AO sequence completion model and Kaggle training pipeline - #96
Closed
Shlok-Parekh09 wants to merge 8 commits into
Closed
Shlok-Parekh09 wants to merge 8 commits into
Shlok-Parekh09 wants to merge 8 commits into
Conversation
…overflow (CodeQL) (PxA-Labs#90)
Shlok-Parekh09
requested review from
archittmittal,
purvanshjoshi and
sr-857
as code owners
August 19, 2026 18:58
Author
|
What about the review anything else to change |
purvanshjoshi
approved these changes
Aug 30, 2026
purvanshjoshi
left a comment
Contributor
There was a problem hiding this comment.
This is a fantastic contribution. The implementation of the Slope-Domain Sequence Completion AI and the Kaggle training pipeline is well-engineered and thoroughly documented. The inclusion of physical constraints and the C integration architecture are particularly well thought out. The tests demonstrate solid improvements over the baseline. Approved.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Description
This Pull Request introduces the Slope-Domain Sequence Completion AI, which acts as a graceful fallback mechanism for the AO system. When the external frame-quality gate detects partial signal loss (e.g., from mild fog, scattering, or occlusion), this model reconstructs the missing or corrupted wavefront slopes ($\Delta x$ , $\Delta y$ ) per sub-aperture based on the temporal consistency of past frames.
Key additions:
SlopeCompletionLSTMModel: A mask-conditioned LSTM architecture with a persistence-prior residual head.predictive_ao_complete_slopes()API for ONNX Runtime integration with zero-dependency static builds.Related Issues
Closes #94
(Integrates with the umbrella AI fallback issue #90 and frame-quality gate #93)
Mathematical/Scientific Details
Verification & Testing Done
Detail the testing performed to verify changes:
test_slope_completion.pypassed 19/19 checks)- Inference Latency: Median ONNX Runtime CPU inference runs at
~1.1 ms.- Accuracy: Achieved an RMSE of
0.5201, reflecting a ~30% reduction in error compared to the baseline spatial interpolation (0.7419).Checklist
Linux kernelfor C/CUDA,PEP 8for Python).docs/,CONTRIBUTOR.md) and added Doxygen comments to public API functions.CHANGELOG.mdwith a summary of my changes.