large-scale text categorization by batch mode active learning steven c.h. hoi †, rong jin ‡,...

45
Large-Scale Text Categorization By Batch Mode Active Learning Steven C.H. Hoi, Rong Jin, Michael R. Lyu† CSE Department, Chinese University of Hong Kong ‡ CSE Department, Michigan State University 26-May, 2006 To appear in International World Wide Web conference, Edinburgh, Scotland, 22-26 May, 2006.

Post on 22-Dec-2015

213 views

Category:

Documents


0 download

TRANSCRIPT

Large-Scale Text Categorization ByBatch Mode Active Learning

Steven C.H. Hoi†, Rong Jin‡, Michael R. Lyu†

† CSE Department, Chinese University of Hong Kong‡ CSE Department, Michigan State University

26-May, 2006

To appear in International World Wide Web conference, Edinburgh, Scotland, 22-26 May, 2006.

2

Outline

Introduction Related Work Batch Mode Active Learning

Theoretical Foundation Convex Optimization Formulation Eigen Space Simplification Bound Optimization Algorithm

Experimental Results Conclusion and Future Work

3

Introduction Text Categorization

Problem Assign documents to predefined topics

Significances Core Web data mining technique Applications: category browsing, vertical search, etc.

Challenges To build efficient classifiers To minimize human labeling effort

4

Introduction Logistic Regression

Efficiency for Training and Prediction Natural Probability Output State-of-the-art performance, etc… Linear model

where is the class label. Simplified notation:

5

Introduction Active Learning

To find most informative unlabeled examplesTraditional Methodology

Choose one unlabeled example for labeling Retrain the classifier with the additional example

Limitation Only one example in each iteration, huge

retraining cost

Our solution: Batch Mode Active Learning To find a batch of most informative unlabeled

examples

6

Outline

Introduction Related Work Batch Mode Active Learning

Theoretical Foundation Convex Optimization Formulation Eigen Space Simplification Bound Optimization Algorithm

Experimental Results Conclusion and Future Work

7

Related Work

Statistical Models for Classification K Nearest Neighbors (Masand et al., SIGIR’92), Decision

Trees (Apte et al., TOIS’94), Bayesian Classifiers (Tzeras et al., SIGIR’93), Inductive Rule Learning (Cohen et al., ICML’95), etc.

Neural Networks (Ruiz et al., IR’02), Support Vector Machines (SVM) (Joachims, ECML’98, Tong et al., ICML’00), and Logistic Regressions (Zhang et al., ICML’00), etc.

8

Related Work

Active Learning Query-By-Committee (Liere et al AAAI’97), EM & Active

Learning (Nigam et al’98), etc.

Margin Based Methods: Support Vector Machine Active Learning (Tong et al., ICML’00)

Measure uncertainty by the distances from decision boundaries

9

Batch Mode Active Learning

(a) Binary classification example (b) Margin-based active learning (c) Batch mode active learning

– Positive examples of class-1

– Negative examples of class-2

– Unlabeled examples

– Selected examples for labeling

Toy Example

D1

D2

10

Outline

Introduction Related Work Batch Mode Active Learning

Theoretical Foundation Convex Optimization Formulation Eigen Space Simplification Bound Optimization Algorithm

Experimental Results Conclusion and Future Work

11

Theoretical Foundation Main Idea:

Based on the theoretical framework of maximization of Fisher information

Problem SettingIn a probabilistic classification framework, assume the classification model is a semi-parametric form

For example, the logistic regression model:

12

Theoretical Foundation The problem of batch mode active learning can be

regarded as a problem to seek a resample distribution q(x) of the unlabeled data.

The examples with large resampling probabilities will be selected as the most informative ones for labeling.

According to statistical estimation theory, active learning should consider a resample distribution q(x) that maximizes the following Fisher information

13

Theoretical Foundation The maximization of Fisher information is equivalent to find

the resample distribution q(x) that minimizes the ratio of two Fisher information matrixes:

For the logistic regression model, the Fisher information matrix can be expressed as:

We replace the integration in the above equation with the summation over the unlabeled data:

14

Convex Optimization Formulation Rewrite the objective function as

Introduce a slack matrix ,then turn the original problem into the following optimization:

In the above, we use

15

Convex Optimization Formulation

By the Schur complementary theorem, i.e.,

we turn it into the following optimization :

16

Convex Optimization Formulation The final optimization problem can be expressed

The above problem belongs to the family of Semi-definite programming (SDP) and can be solved by convex optimization techniques.

17

Eigen Space Simplification

Directly solving the above optimization problem is computationally expensive for the large-size slack matrix variable of M.

In order to reduce the computational complexity, we propose an Eigen space simplification method to make the solution simpler and more effective.

We assume that M is expanded in the Eigen space of the Fisher information matrix Ip.

18

Eigen Space Simplification

Let be the top s eigen vectors of the Fisher information matrix Ip, where λ1 ≥ λ2 ≥ . . . ≥ λs, then we assume the matrix M has the following form:

We rewrite the inequality

19

Eigen Space Simplification

Using the eigen expression, we have

Given the necessary condition for is

Therefore, we have the following result

20

Eigen Space Simplification

The above necessary condition leads to following constraints:

Meanwhile, the objective function of tr(M) can be expressed as

21

Eigen Space Simplification By putting the above two expressions together, we

transform the SDP problem into the following approximate optimization problem:

Note that the above optimization problem belongs to convex optimization since f(x) = 1/x is convex when x ≥ 0.

22

Bound Optimization Algorithm

Lemma 1: Let L(q) be the objective function,

we have the following conclusion:

23

Bound Optimization Algorithm

Given the lemma 1, now instead of optimizing the original objective function L(q), we can optimize its upper bound using simple updating equations:,

This algorithm will guarantee to converge to a local optimal. Since the original problem is a convex optimization problem, the above updating procedure will guarantee to converge to a global optimal.

24

Bound Optimization Algorithm

The updating step:

Some Observations (i) The example with a large classification uncertainty

will be assigned with a large probability.

(ii) The example that is similar to many unlabeled examples is more likely to be selected.

25

Outline

Introduction Related Work Batch Mode Active Learning

Theoretical Foundation Convex Optimization Formulation Eigen Space Simplification Bound Optimization Algorithm

Experimental Results Conclusion and Future Work

26

Experimental Testbeds 3 standard text datasets

Reuters-21578 dataset (10788) Two web-related datasets:

WebKB (4518) and Newsgroup (10966)

27

Experimental Settings

A standard feature selection by Information Gain is conducted to remove uninformative features, in which 500 of the most informative features are selected.

The F1 metric is adopted as our evaluation metric, which has been shown to be more reliable metric than other metrics such as the classification accuracy. More specifically, the F1 is defined as

where p and r are precision and recall. Parameters of LogReg and SVM are determined by a

standard cross validation method.

28

Comparison Schemes Two popular active learning methods:

SVM-AL: the classification uncertainty of an example x is determined by its distance to the decision boundary

The smaller the distance d(x;w, b) is, the more the classification uncertainty will be.

LogReg-AL: the logistic regression active learning algorithm that measures the classification uncertainty based on the entropy of the distribution p(y|x).

The larger the entropy of x is, the more uncertain we are about the class labels of x.

Our Batch Mode Active Learning algorithm with logistic regression, i.e., LogReg-BMAL in short.

29

Empirical Evaluation Experimental Results with Reuters-21578

average results over 40 executions 100 training examples and 100 active examples

30

Empirical Evaluation Experimental Results with Reuters-21578

31

Empirical Evaluation Experimental Results with Reuters-21578

32

Empirical Evaluation Experimental Results with Web-KB Dataset

33

Empirical Evaluation Experimental Results with Newsgroup Dataset

34

Conclusion A batch mode active learning scheme is proposed to

attack the challenge of large-scale text categorization. The main contributions include

A new active learning scheme is suggested for large-scale text categorization to overcome the limitation of traditional active learning;

A batch mode active learning solution is formulated by convex optimization techniques;

An effective bound optimization algorithm is proposed to solve the batch mode active learning problem

Extensive experiments are conducted for empirical evaluations in comparisons with state-of-the-art active learning approaches for text categorization

35

Future Work

To combine batch mode active learning with semi-supervised learning

To improve the computational costs To study the convergence of the bound optimization To extend the methodology for other classification models

36

Thank you for your attention!

Questions?

Http://www.cse.cuhk.edu.hk/~chhoi/

37

Appendix A – Statistical Estimation Theory

Given a semi-parametric model, say the logistic model as:

In theory, one can use the maximum-likelihood estimate (MLE) to determine the model parameter as:

In theory, the MLE achieves the Cramer-Rao lower bound, thus, the MLE is the asymptotically most efficient estimator, whose efficiency can be measured by the Fisher information that is intrinsic to the probability model.

38

Appendix A – Statistical Estimation Theory (cont.)

More specifically, the expected log-likelihood to measure the goodness of q(x) can be given as

Hence, according to the Crammer-Rao lower bound, the MLE based on the resample distribution q(x) that minimizes is the most efficient estimator of alpha among all estimators based on a resampling of x.

Therefore, the result of q to solve the optimization is the optimal sample distribution for active learning.

39

Appendix B.

Fisher Information and Cramer-Rao lower bound Fisher information is thought of as the amount of information that an

observable random variable X carries about an unobservable parameter θ upon which the probability distribution of X depends. Since the expectation of the score is zero, the variance is also the second moment of the score and so the Fisher information can be written

In statistics, the Cramér-Rao lower bounds express a lower bound on the accuracy of a statistical estimator, based on Fisher information.

It states that the reciprocal of the Fisher information, , of a parameter θ, is a lower bound on the variance of an unbiased estimator of the parameter (denoted ).

40

Appendix C – Convexity Theorem

Theorem. Any locally optimal point of a convex problem is (globally) optimal.

41

42

Appendix D – Semidefinite Programming (SDP)

43

Appendix E: Proof of Lemma1

Lemma 1: Let L(q) be the objective function in (15), we have the following conclusion

Proof.

44

Proof (cont.):Using the convexity property of reciprocal function, namely

for x ≥ 0 and p.d.f. .

We can arrive the following deduction

45

Proof (cont.):Substituting the above inequation back to L(q), we can attain the following inequality:

This finishes the proof of the inequality lemma. □