File tree Expand file tree Collapse file tree 1 file changed +16
-1
lines changed Expand file tree Collapse file tree 1 file changed +16
-1
lines changed Original file line number Diff line number Diff line change @@ -87,7 +87,22 @@ class _CliqueData(NamedTuple):
87
87
88
88
89
89
class LabelModel (nn .Module ):
90
- """A conditionally independent LabelModel to learn LF weights and assign training labels.
90
+ r"""A model for learning the LF accuracies and combining their output labels.
91
+
92
+ This class learns a model of the labeling functions' conditional probabilities
93
+ of outputting the true (unobserved) label `Y`, `P(\lf | Y)`, and uses this learned
94
+ model to re-weight and combine their output labels.
95
+
96
+ This class is based on the approach in [Training Complex Models with Multi-Task
97
+ Weak Supervision](https://arxiv.org/abs/1810.02840), published in AAAI'19. In this
98
+ approach, we compute the inverse generalized covariance matrix of the junction tree
99
+ of a given LF dependency graph, and perform a matrix completion-style approach with
100
+ respect to these empirical statistics. The result is an estimate of the conditional
101
+ LF probabilities, `P(\lf | Y)`, which are then set as the parameters of the label
102
+ model used to re-weight and combine the labels output by the LFs.
103
+
104
+ Currently this class uses a conditionally independent label model, in which the LFs
105
+ are assumed to be conditionally independent given `Y`.
91
106
92
107
Examples
93
108
--------
You can’t perform that action at this time.
0 commit comments