Hypervolume Knowledge Gradient (#1950)

Summary: Pull Request resolved: #1950 This includes support for decoupled evaluations and multi-fidelity optimization Reviewed By: Balandat Differential Revision: D47809667 fbshipit-source-id: 7e7a423e279f4ba3225d74ce90e7abae65ebfb4d
pytorch · Nov 10, 2023 · dad72e9 · dad72e9
1 parent 9bab1d9
commit dad72e9
Show file tree

Hide file tree

Showing 4 changed files with 997 additions and 0 deletions.
diff --git a/botorch/acquisition/multi_objective/__init__.py b/botorch/acquisition/multi_objective/__init__.py
@@ -8,6 +8,10 @@
     ExpectedHypervolumeImprovement,
     MultiObjectiveAnalyticAcquisitionFunction,
 )
+from botorch.acquisition.multi_objective.hypervolume_knowledge_gradient import (
+    qHypervolumeKnowledgeGradient,
+    qMultiFidelityHypervolumeKnowledgeGradient,
+)
 from botorch.acquisition.multi_objective.max_value_entropy_search import (
     qMultiObjectiveMaxValueEntropy,
 )
@@ -35,6 +39,8 @@
     "get_default_partitioning_alpha",
     "prune_inferior_points_multi_objective",
     "qExpectedHypervolumeImprovement",
+    "qHypervolumeKnowledgeGradient",
+    "qMultiFidelityHypervolumeKnowledgeGradient",
     "qNoisyExpectedHypervolumeImprovement",
     "MOMF",
     "qMultiObjectiveMaxValueEntropy",