Inference

[NastyNastya2024]

Could you point me to the specific parts of the codebase that handle inference in the context of vertical federated learning? If inference is not yet implemented, are there any ongoing discussions or plans for its inclusion?

We are particularly interested in performing inference for a gradient boosting model. If inference is not yet implemented, are there any ongoing discussions or plans for its inclusion?"

[trivialfis]

This class handles column-split inference:

xgboost/src/predictor/cpu_predictor.cc

Line 418 in 5fbab40

class ColumnSplitHelper {

The vertical federated learning can handle inference and evaluation:

xgboost/python-package/xgboost/testing/federated.py

Line 79 in 5fbab40

watchlist = [(dtest, "eval"), (dtrain, "train")]

But the feature is overall still a work in progress; no active contributor is working on it at the moment.

Related:

• Vertical Federated Learning RFC #8424
• In-memory inputs for column split and vertical federated learning #9619

[NastyNastya2024]

Thank you for your responce - could you provide some information if there is Inference in NV Flare NVIDIA infrastracture itself ?

[trivialfis]

Please reach out to nvflare for related questions. I haven't been catching up with the development in nvflare lately.