Proposal: Add Split Learning Example to Flower #5732
Description
Type
Documentation Improvement
Description
Proposed Contribution
I’d like to contribute a minimal working example of split learning under examples/split_learning/.
The example would:
- Use PyTorch for model definition.
- Split the model into client-side and server-side parts.
- Demonstrate training on a simple dataset (e.g., CIFAR-10 or MNIST).
- Show how Flower’s Strategy + Client APIs can orchestrate split learning.
Planned Implementation
Implementation Plan
- Implement a
SplitLearningStrategyextending Flower’sStrategy. - Implement a
SplitClientthat handles cut-forward / cut-backward operations. - Add a runnable demo in
examples/split_learning/with a README. - Add tests (basic correctness + CI integration).
- Provide a Colab/Jupyter notebook tutorial for users.
Additional Context
If maintainers agree with this proposal, I’d like to begin implementing it and open a PR.