(Bunescu and Pasca, 2006) propose to generalize beyond context-entity correlation s(d, e) with word-category correlation s(w, c).
this method works at word level, and does not scale well to large number of categories.