Source-linked AI summary

Semi-Supervised Joint Estimation of Word and Document Readability

Yoshinari Fujinuma, Masato Hagiwara

arXiv:2104.13103v1cs.CL

TL;DR

Readability estimation has often treated words and documents separately, despite their recursive relationship and the limited availability of annotated resources. The paper jointly estimates both with a semi-supervised graph convolutional network, which achieves higher accuracy and remains effective with less labeled data. Its scope is limited by leaving richer node types for future work and by assuming a normal distribution for the difficulty estimate.

  • Problem

    Existing word and document readability methods often depend on annotated resources for the other level, while such resources are limited and costly to build.

    Source: §1 · p. 1
  • Method

    A semi-supervised GCN models words and documents as graph nodes and recursively infers their embeddings to jointly estimate their readability.

    Source: §1 · pp. 1–2, §3.2 · p. 3
  • Results

    The GCN achieves higher accuracy by capturing the recursive difficulty relationship between words and documents, even with a smaller amount of labeled data.

    Source: §1 · p. 2, §5 · p. 4
  • Takeaways & Limitations

    The graph-based formulation provides effective complementary structure for readability estimation and can incorporate diverse node types such as subwords, paragraphs, and grammatical concepts.

    Source: §4.2 · p. 4, §5 · p. 4

    The paper leaves incorporating diverse node types for future work and assumes the difficulty estimate is normally distributed when mapping CEFR levels.

    Source: §3.2 · p. 2, §5 · p. 5

Abstract

from arXiv · show

Readability or difficulty estimation of words and documents has been investigated independently in the literature, often assuming the existence of extensive annotated resources for the other. Motivated by our analysis showing that there is a recursive relationship between word and document difficulty, we propose to jointly estimate word and document difficulty through a graph convolutional network (GCN) in a semi-supervised fashion. Our experimental results reveal that the GCN-based method can achieve higher accuracy than strong baselines, and stays robust even with a smaller amount of labeled data.

1 Introduction

Word and document readability are closely intertwined, yet existing methods often estimate them separately and depend on annotated resources for the other level. The paper proposes a semi-supervised GCN that jointly estimates both by exploiting their recursive difficulty relationship.

Motivation

  • Existing readability methods often require annotated lexical or textual resources, while such resources remain scarce outside major European languages and costly to construct.Resources have been created for languages including English, French, and Swedish, but manually building them demands linguistic expertise and effort.Source: §1 · p. 1
  • Word and document difficulty are recursively related: document difficulty correlates with its hardest word, while word difficulty correlates with the easiest document containing it.This bidirectional relationship motivates estimating both levels together rather than treating them independently.Source: §1 · p. 1

Proposed approach

  • The proposed method jointly estimates word and document readability semi-supervisedly by modeling words and documents as graph nodes and recursively inferring their embeddings with GCN layers.The approach uses both labeled supervision and the recursive word-document relationship.Source: §1 · pp. 1–2

Contributions

  • GCNs effectively exploit unlabeled data and remain effective when fewer labeled examples are available.The paper reframes readability estimation as a semi-supervised joint problem and identifies GCNs as a solution for this setting.Source: §1 · p. 2

2 Task Definition

The task is to map words and documents to difficulty labels using both labeled and unlabeled instances. Documents are treated as arbitrary-length texts, with paragraphs used as the basic prediction unit.

Task formulation

  • Joint readability estimation maps words and documents to difficulty labels, supporting either classification or regression formulations.The classification setting uses six CEFR levels from A1 to C2, while regression produces a real-valued readability estimate.Source: §2 · p. 2

Semi-supervised setting

  • Words and documents are partitioned into labeled and unlabeled subsets, with inference using labeled supervision and potentially word-document relationships among unlabeled instances.The labeled subsets are W_L and D_L, while W_U and D_U provide additional structure.Source: §2 · p. 2

3 Exploiting Recursive Relationship by Graph Convolutional Networks

The paper models words and documents as nodes in a heterogeneous graph because their difficulty is recursively related. A GCN propagates information through this graph, uses separate prediction layers for words and documents, and incorporates unlabeled evaluation nodes.

3.1 Recursive Relationship of Word and Document Difficulty

  • Word difficulty correlates with the minimum difficulty of documents containing the word, while document difficulty correlates with the maximum difficulty of its words.This recursive relationship motivates joint estimation rather than treating word and document readability independently.Source: §3.1 · p. 2, Figure 2 · p. 2

3.2 Graph Convolutional Networks on Word-Document Graph

  • The GCN treats words and documents as graph nodes whose hidden layers recursively connect them to capture potentially nonlinear readability relationships.Unlabeled evaluation nodes are included because GCN inference is transductive.Source: §3.2 · pp. 2–3
  • The graph uses TF-IDF document-word edges, normalized PMI word-word edges, and self-loops to encode heterogeneous relationships.The adjacency matrix is symmetrically normalized using node degrees before convolution.Source: §3.2 · pp. 2–3
  • Separate final linear layers predict word and document difficulty, while a weighted combination of their losses trains the joint model.The framework supports regression with mean squared error and classification with softmax cross-entropy.Source: §3.2 · p. 3
  • A simplified linear GCN performed worse than the ReLU hidden-layer model, and a shared final linear layer also underperformed.These comparisons support using nonlinear hidden layers and separate output layers for the two node types.Source: §3.2 · p. 3

4 Experiments

Experiments compare GCN-based joint readability estimation with regression, classification, boosted-tree, and attention baselines on CEFR-annotated word and document resources. GCN improves both word and document accuracy, remains robust with less labeled data, and benefits from graph structure and BERT document embeddings.

4.1 Results

  • GCNs improve both document and word accuracy over the baselines in the reported difficulty-estimation results.Table 2 reports accuracy and correlation for logistic regression and GCN classification and regression variants.Source: §4.1 · p. 4, Table 2 · p. 4
  • GCN propagates an A1 document label to its word node, correctly identifying “bicycle” where logistic regression predicts A2 from input features alone.Source: §4.1 · p. 4

4.2 Ablation Study on Features

  • Without traditional or embedding features, GCN achieves results comparable to some baselines, showing that word-document graph structure supplies complementary readability signals.The ablation’s “None” setting uses one-hot node features and relies solely on graph structure.Source: §4.2 · p. 4, Table 3 · p. 4
  • Removing BERT embeddings decreases document accuracy by 0.112 on the Cambridge Readability Dataset.The result supports BERT embeddings as a powerful document-readability feature in this setting.Source: §4.2 · p. 4

4.3 Training on Less Labeled Data

  • GCN performance remains robust with smaller training sets, especially for word accuracy, by exploiting recursive word-document relationships and graph context.The word-accuracy advantage over the baseline becomes larger than the document-accuracy gap when training data is small.Source: §4.3 · p. 4

5 Conclusion

The proposed GCN jointly estimates word and document readability, achieving higher accuracy while remaining robust with smaller amounts of labeled data. Figure 3 examines word and document accuracy as training-data volume varies, while future work considers broader node types.

  • The GCN jointly estimates word and document readability and achieves higher accuracy while using smaller amounts of labeled data.Its advantage is attributed to capturing the recursive difficulty relationship between words and documents.Source: §5 · p. 4, Figure 3 · p. 5
  • Figure 3 compares word and document accuracy across different amounts of training data.Source: Figure 3 · p. 5
  • The framework could be extended to include subwords, paragraphs, and grammatical concepts, which the paper leaves for future work.Source: §5 · p. 5

A Hyperparameter Details

The study selects GCN hyperparameters separately for document and word difficulty estimation in classification and regression settings. The selected configurations vary in dropout, epochs, hidden units, layers, and PMI window width.

Search procedure

  • The hyperparameter search samples 200 combinations across α, learning rate, dropout, epochs, hidden units, layers, and PMI window width.Source: A Hyperparameter Details · p. 6

Classification setting

  • Classification uses separate GCN configurations for document and word difficulty estimation.Document estimation uses dropout 0.5, 500 epochs, 512 hidden units, 2 layers, and PMI window width 5; word estimation uses dropout 0.2, 250 epochs, 64 hidden units, 1 layer, and disabled PMI.Source: A Hyperparameter Details · p. 6

Regression setting

  • Regression also uses separate GCN configurations for document and word difficulty estimation.Document estimation uses dropout 0.3, 1500 epochs, 128 hidden units, 2 layers, and PMI window width 5; word estimation uses dropout 0.1, 500 epochs, 512 hidden units, 2 layers, and disabled PMI.Source: A Hyperparameter Details · p. 6
Loading 2104.13103v1…