data mining final project nick foti eric kee. topic: author identification author identification...
Post on 21-Dec-2015
216 views
TRANSCRIPT
Topic: Author Identification• Author Identification
– Given writing samples, can we determine who wrote them?
– This is a well studied field• See also: “stylometry”
– This has been applied to works such as• The Bible• Shakespeare• Modern texts as well
Corpus Design• A corpus is:
– A body of text used for linguistic analysis
• Used Project Gutenberg to create corpus• The corpus was designed as follows
– Four authors of varying similarity• Anne Brontë• Charlotte Brontë• Charles Dickens• Upton Sinclair
– Multiple books per author
• Corpus size: 90,000 lines of text
Dataset Design• Extracted features common in literature
– Word Length– Frequency of “glue” words
• See Appendix A and [1,2] for list of glue words
• Note: corpus was processed using– C#, Matlab, Python
• Data set parameters are– Number of dimensions: 309
• Word length and 308 glue words
– Number of observations: ≈ 3,000• Each obervation ≈ 30 lines of text from a book
Classifier Testing and Analysis• Tested classifier with test data
– Used testing and training data sets • 70% for training, 30% for testing
– Used cross-validation
• Analyzed Classifier Performance– Used ROC plots– Used confusion matrices
• Used common plotting scheme (right)
Anne B.TP
Anne B.FP
Charlotte B.TP
Charlotte B.FP
78%
22%
55%
45%
E X A M P L E
Red Dots Indicate True-Positive Cases
Word Length Classification• Calculated average word length for each
observation
• Computed gaussian kernel density from word length samples
• Used ROC curve to calculate cutoff– Optimized sensitivity and specificity with equal
importance
Word Length: Anne B. vs Upton S.
100%
0%
Anne B.T P
Anne B.F P
Upton SinclairT P
Upton SinclairF P
100%
0%
Anne BrontëCharlotte Brontë
Word Length: Brontë vs. Brontë
100%
0%
Anne B.T P
Anne B.F P
Charlotte B.T P
Charlotte B.F P
78.1%
21.9%
Anne BrontëCharlotte Brontë
Principal Component Analysis• Used PCA to find a better axis
• Notice: distribution similar to word length distribution
• Is word lengththe only usefuldimension?
Anne Brontë vs. Upton Sinclair
Word Length Density
PCA Density
Without word lengthPrincipal Component Analysis
• It appears that word length is the most useful axis
• We’ll come back to this…
Anne Brontë vs. Upton Sinclair
PCA Density
K-Means• Used K-means to find dominant patterns
– Unnormalized – Normalized
• Trained K-means on training set• To classify observations in test set
– Calculate distance of observation to each class mean
– Assign observation to the closest class
• Performed cross-validation to estimate performance
Unnormalized K-means Anne Brontë vs. Upton Sinclair
98.1%
1.9%
Anne B.T P
Anne B.F P
Upton SinclairT P
Upton SinclairF P
92.1%
7.9%
Unnormalized K-means Anne Brontë vs. Charlotte Brontë
95.7%
4.3%
Anne B.T P
Anne B.F P
Charlotte B.T P
Charlotte B.F P
74.7%
25.3%
Normalized K-means Anne Brontë vs. Upton Sinclair
53.3%
46.7%
Anne B.T P
Anne B.F P
Upton SinclairT P
Upton SinclairF P
49.4%50.6%
Normalized K-means Anne Brontë vs. Charlotte Brontë
15.8%
84.2%
Anne B.T P
Anne B.F P
Charlotte B.T P
Charlotte B.F P
86.7%
13.3 %
Discriminant Analysis• Peformed discriminant analysis
– Computed with equal covariance matrices• Used average Omega of class pairs
– Computed with unequal covariance matrices• Quadratic discrimination fails because covariance
matrices have 0 determinant (see equation below)
– Computed theoretical misclassification probability
• To perform quadratic discriminant analysis– Compute Equation 1 for each class– Choose class with minimum value
(1)
Discriminant Analysis
Anne Brontë vs. Upton Sinclair
92.2%
3.8%
Anne B.T P
Anne B.F P
Upton SinsclairT P
Upton SinclairF P
96.2%
7.8%
Theoretical P(err) = 0.149Empirical P(err) = 0.116
Discriminant Analysis
Anne Brontë vs. Charlotte Brontë
92.7%
7.3%
Anne B.T P
Anne B.F P
Charlotte B.T P
Charlotte B.F P
89.2%
10.8%
Theoretical P(err) = 0.181Empirical P(err) = 0.152
Logistic Regression• Fit linear model to training data on all dimensions• Threw out singular dimensions
– Left with ≈ 298 coefficients + intercept
• Projected training data onto synthetic variable– Found threshold by minimizing error of misclassification
• Projected testing data onto synthetic variable– Used threshold to classify points
Logistic Regression
Anne BTP
Anne BTP
Charlotte BTP
Charlotte BTP
Anne BrontëCharlotte Brontë
89.5%
10.5% 8%
92%
Anne Brontë vs Charlotte Brontë
Logistic Regression
Anne BTP
Anne BFP
Upton STP
Upton SFP
Anne BrontëUpton Sinclair
98%
2%
99%
2%
Anne Brontë vs Upton Sinclair
4-Class K-means• Used K-means to find patterns among all
classes– Unnormalized – Normalized
• Trained using a training set
• Tested performance as in 2-class K-means
• Performed cross-validation to estimate performance
Unnormalized K-Means
CD
TP
AB
FP
CB
FP
US
FP
CD
FP
AB
TP
CB
FP
US
FP
CD
FP
AB
FP
CB
TP
US
FP
CD
FP
AB
FP
CB
FP
US
FP
22%
54%
87%
34%
59%
88%
Anne Brontë Charlotte Brontë Upton SinclairCharles Dickens
4-Class Confusion Matrix
Normalized K-Means
CD
TP
AB
FP
CB
FP
US
FP
CD
FP
AB
TP
CB
FP
US
FP
CD
FP
AB
FP
CB
TP
US
FP
CD
FP
AB
FP
CB
FP
US
FP
20%
67%
26%
67%70% 67%
27%
Anne Brontë Charlotte Brontë Upton SinclairCharles Dickens
4-Class Confusion Matrix
Additional K-means testing• Also tested K-means without word length
– Recall that we had perfect classification with 1D word length (see plot below)
– Is K-means using only 1 dimension to classify?
Note: perfect classification only occurs between Anne B. and Sinclair
Anne Brontë vs. Upton Sinclair
CD
TP
AB
FP
CB
FP
US
FP
CD
FP
AB
TP
CB
FP
US
FP
CD
FP
AB
FP
CB
TP
US
FP
CD
FP
AB
FP
CB
FP
US
FP
35%
29%
44%
33%35%
43%
72%
Anne Brontë Charlotte Brontë Upton SinclairCharles Dickens4-Class Confusion Matrix
Unnormalized K-Means (No Word Length)
• K-means can classify without word length
Multinomial Regression• Multinomial distribution
– Extension of binomial distribution• Random variable is allowed to take on n values
• Used multinom(…) to fit log-linear model for training– Used 248 dimensions (max limit on computer)– Returns 3 coefficients per dimension and 3
intercepts
• Found probability that observations belongs to each class
Multinomial Regression• Multinomial Logit Function is
where j are the coefficients and cj are the intercepts
• To classify– Compute probabilities
• Pr(yi = Dickens), Pr(yi = Anne B.), …
– Choose class with maximum probability
Multinomial Regression
CD
TP
AB
FP
CB
FP
US
FP
CD
FP
AB
TP
CB
FP
CB
FP
CD
FP
AB
FP
CB
TP
US
FP
CD
FP
AB
FP
CB
FP
US
FP
78% 86% 83% 93%
Anne Brontë Charlotte Brontë Upton SinclairCharles Dickens
4-Class Confusion Matrix
CD
TP
AB
FP
CB
FP
US
FP
CD
FP
AB
TP
CB
FP
CB
FP
CD
FP
AB
FP
CB
TP
US
FP
CD
FP
AB
FP
CB
FP
US
FP
76% 79% 79% 91%
Upton SinclairCharles Dickens Anne Brontë Charlotte Brontë
(Without Word Length)
4-Class Confusion Matrix
Multinomial Regression
• Multinomial regression does not require word length
Appendix A: Glue WordsI a aboard about above across after again against ago ahead all almost along alongside already also although always am amid amidst among amongst an and another any anybody anyone anything anywhere apart are aren't around as aside at away back backward backwards be because been before beforehand behind being below between beyond both but by can can't cannot could couldn't dare daren't despite did didn't directly do does doesn't doing don't done down during each either else elsewhere enough even ever evermore every everybody everyone everything everywhere except fairly farther few fewer for forever forward from further furthermore had hadn't half hardly has hasn't have haven't having he hence her here hers herself him himself his how however if in indeed inner inside instead into is isn't it its itself just keep kept later least less lest like likewise little low lower many may mayn't me might mightn't mine minus more moreover most much must mustn't my myself near need needn't neither never nevertheless next no no-one nobody none nor not nothing notwithstanding now nowhere of off often on once one ones only onto opposite or other others otherwise ought oughtn't our ours ourselves out outside over own past per perhaps please plus provided quite rather really round same self selves several shall shan't she should shouldn't since so some somebody someday someone something sometimes somewhat still such than that the their theirs them themselves then there therefore these they thing things this those though through throughout thus till to together too towards under underneath undoing unless unlike until up upon upwards us versus very via was wasn't way we well were weren't what whatever when whence whenever where whereas whereby wherein wherever whether which whichever while whilst whither who whoever whom with whose within why without will won't would wouldn't yet you your yours yourself yourselves
Conclusions• Authors can be identified by their word usage frequencies• Word length may be used to distingush between the Brontë
sisters– Word length does not, however, extend to all authors (See Appendix C)
• The glue words describe genuine differences between all four authors– K-means distinguishes the same patterns that multinomial regression
classifies• This indicates that supervised training finds legitimate patterns, rather than
artifacts
• The Brontë sisters are the most similar authors• Upton Sinclair is the most different author
Appendix C: Single Dimension 4-Author Classification
CD
TP
AB
FP
CB
FP
US
FP
CD
FP
AB
TP
CB
FP
US
FP
CD
FP
AB
FP
CB
TP
US
FP
CD
FP
AB
FP
CB
FP
US
FP
22%
46%
94%
6%
11%
54%
96%
Anne Brontë Charlotte Brontë Upton SinclairCharles Dickens4-Class Confusion Matrix
3%
Classification using Multinomial Regression