Skip to content

Commit

Permalink
Adds a citation detector
Browse files Browse the repository at this point in the history
** Why are these changes being introduced:

A certain percentage of our search traffic is made up of formal
citations to existing works, in a variety of formats. It would be good
to have a detector to identify these and pluck them out consistently for
further work (reconciliation, re-formatting, etc)

** Relevant ticket(s):

* https://mitlibraries.atlassian.net/browse/tco-97

Also TCO-96 and TCO-95 get some benefit from this.

** How does this address that need:

This adds a new Detector::Citation class, with attendant changes to the
seeds and test fixtures. This is a different type of detector than we've
written in the past, using a multi-layer approach that first compiles
some discrete small information using regexes and counts, which are then
assessed by a second routine that calculates a final score. Terms which
score high enough can have a Detection registered using our usual
workflow.

The smaller discrete signals were designed after looking over examples
of five different citation formats: MLA, APA, Chicago, Terabian, and
IEEE. Examples of these patterns include formats for volume, issue, page
ranges, quoted titles, and name formatting. These are implemented using
regular expressions.

A second set of discrete signals are generated using counts, by looking
at how many characters, words, and specific symbols are found in the
search string (commas, periods, and other potential separators). Each of
these counts are compared to a threshold value, so that if enough of
them are in the term then the citation score gets raised.

While I feel okay about the overall structure of this detector, the
specific thresholds I'm using probably need to be verified against real
world data. I have some ideas about how to pursue this in the future, as
a refinement ticket later on.

** Document any side effects to this change:

* While there are similarities between this detector and the structure
of the StandardIdentifiers detector, I've chosen to vary some parts of
the approach as well (using scan rather than match, for example, or
defining the regexes using a constant). Ultimately I think we should
probably have a standardized approach, but for now I think some
variation might help us compare and contrast between them.
  • Loading branch information
matt-bernhardt committed Oct 10, 2024
1 parent 7d6a2e3 commit ed39b74
Show file tree
Hide file tree
Showing 7 changed files with 411 additions and 0 deletions.
156 changes: 156 additions & 0 deletions app/models/detector/citation.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,156 @@
# frozen_string_literal: true

class Detector
# Detector::Citation attempts to identify citations based on the prevalence of individual sub-patterns. It is not
# targeted at a particular citation format, but was designed based on characteristics of five formats: APA, MLA,
# Chicago, Terabian, and IEEE.
#
# It receives a Term object, which is parsed in various ways en route to calculating a final score. Terms with a
# higher score are more citation-like, while a score of 0 indicates a Term that has no hallmarks of being a citation.
# Terms whose score is higher than the REQUIRED_SCORE value can be registered as a Detection.
class Citation
attr_reader :score, :subpatterns, :summary

# Citation patterns are regular expressions which attempt to identify structures that are part of many citations.
# This object is used as part of the pattern_checker method. Some of these patterns may get promoted to the Detector
# model if they prove useful beyond a Citation context.
CITATION_PATTERNS = {
apa_volume_issue: /\d+\(\d+\)/,
no: /no\.\s\d+/,
pages: /\d+-+\d+/,
pp: /pp\.\s\d+/,
vol: /vol\.\s\d+/,
year_parens: /\(\d{4}\)/,
brackets: /\[.*?\]/,
lastnames: /[A-Z][a-z]+[.,]/,
quotes: /".*?"/
}.freeze

# The required score value is the threshold needed for a Term to be officially recorded with a Detection.
REQUIRED_SCORE = 6

# Summary thresholds are used by the calculate_score method. This class counts the number of occurrences of specific
# characters in the @summary instance variable. The thresholds here determine whether any of those counts are high
# enough to contribute to the Term's citation score.
SUMMARY_THRESHOLDS = {
characters: 25,
colons: 2,
commas: 2,
periods: 2,
semicolons: 2,
words: 5
}.freeze

# The initializer handles the parsing of a Term object, and subsequent population of the @subpatterns, @summary,
# and @score instance variables. @subpatterns contains all the citation components which have been flagged by the
# CITATION_PATTERNS hash. @summary contains counts of how often certain characters or words appear in the Term.
# Finally, the @score value is a summary of how many elements in the subpatterns or summary report were detected.
#
# @note This method can be called directly via Detector::Citation.new(Term). It is also called indirectly via the
# Detector::Citation.record(Term) instance method. This method can be called directly when a Detection is not
# desired.
def initialize(term)
@subpatterns = {}
@summary = {}
pattern_checker(term.phrase)
summarize(term.phrase)
@score = calculate_score
end

# The record method first runs all of the parsers by running the initialize method. If the resulting score is higher
# than the REQUIRED_SCORE value, then a Detection is registered.
#
# @return nil
def self.record(term)
cit = Detector::Citation.new(term)
return unless cit.score >= REQUIRED_SCORE

Detection.find_or_create_by(
term:,
detector: Detector.where(name: 'Citation').first,
detector_version: ENV.fetch('DETECTOR_VERSION', 'unset')
)

nil
end

private

# This combines the two reports generated by the Citation detector (subpatterns and summary), and calculates the
# final score value from their contents.
#
# Any detected subpattern is counted toward the score (multiple detections do not get counted twice). For example,
# if the brackets pattern finds two matches, it still only adds one to the final score.
#
# For the summary report, each value is compared with a threshold value in the SUMMARY_THRESHOLDS constant. The
# number of values which meet or exceed their threshold are added to the score. As an example, if a search term has
# five words, this value is compared to the word threshold (also five). Because the threshold is met, the score gets
# incremented by one.
#
# @return integer
def calculate_score
summary_score = @summary.count do |key, value|
SUMMARY_THRESHOLDS.key?(key) && value >= SUMMARY_THRESHOLDS[key]
end

summary_score + @subpatterns.length
end

# This calculates the number of characters in the search term. It is called by the summarize method.
def characters(term)
term.length
end

# This counts the number of colons that appear in the search term, because they tend to appear more often in
# citations than in other searches. It is called by the summarize method.
def colons(term)
term.count(':')
end

# This counts the number of commas in the search term. It is called by the summarize method.
def commas(term)
term.count(',')
end

# This builds one of the two main components of the Citation detector - the subpattern report. It uses each of the
# regular expressions in the CITATION_PATTERNS constant, extracting all matches using the scan method.
#
# @return hash
def pattern_checker(term)
CITATION_PATTERNS.each_pair do |type, pattern|
@subpatterns[type.to_sym] = scan(pattern, term) if scan(pattern, term).present?
end
end

# This counts the number of periods in the search term. It is called by the summarize method.
def periods(term)
term.count('.')
end

# This is a convenience method for the scan method, which is used by pattern_checker.
def scan(pattern, term)
term.scan(pattern).map(&:strip)
end

# This counts the semicolons in the search term. It is called by the summarize method.
def semicolons(term)
term.count(';')
end

# This builds one of the two main components of the Citation detector - the summary report. It calls each of the
# methods in the first line - which all return integers - and puts the result as a key-value pair in the @summary
# instance variable.
#
# @return hash
def summarize(term)
%w[characters colons commas periods semicolons words].each do |check|
@summary[check.to_sym] = send(check, term)
end
end

# This counts the number of words in the search term. It is called by the summarize method.
def words(term)
term.split.length
end
end
end
1 change: 1 addition & 0 deletions app/models/term.rb
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ class Term < ApplicationRecord
#
# @return nil
def record_detections
Detector::Citation.record(self)
Detector::StandardIdentifiers.record(self)
Detector::Journal.record(self)
Detector::Lcsh.record(self)
Expand Down
6 changes: 6 additions & 0 deletions db/seeds.rb
Original file line number Diff line number Diff line change
Expand Up @@ -32,8 +32,14 @@
Detector.find_or_create_by(name: 'PMID')
Detector.find_or_create_by(name: 'Journal')
Detector.find_or_create_by(name: 'SuggestedResource')
Detector.find_or_create_by(name: 'Citation')

# DetectorCategories
DetectorCategory.find_or_create_by(
detector: Detector.find_by(name: 'Citation'),
category: Category.find_by(name: 'Transactional'),
confidence: 0.3
)
DetectorCategory.find_or_create_by(
detector: Detector.find_by(name: 'DOI'),
category: Category.find_by(name: 'Transactional'),
Expand Down
5 changes: 5 additions & 0 deletions test/fixtures/detector_categories.yml
Original file line number Diff line number Diff line change
Expand Up @@ -38,3 +38,8 @@ six:
detector: lcsh
category: informational
confidence: 0.7

seven:
detector: citation
category: transactional
confidence: 0.3
3 changes: 3 additions & 0 deletions test/fixtures/detectors.yml
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,9 @@
# created_at :datetime not null
# updated_at :datetime not null
#
citation:
name: 'Citation'

doi:
name: 'DOI'

Expand Down
3 changes: 3 additions & 0 deletions test/fixtures/terms.yml
Original file line number Diff line number Diff line change
Expand Up @@ -37,3 +37,6 @@ suggested_resource_jstor:

multiple_detections:
phrase: 'Environmental and Health Impacts of Air Pollution: A Review. Frontiers in Public Health. PMID: 32154200. DOI: 10.3389/fpubh.2020.00014'

citation:
phrase: "A. Altun, &quot;Understanding hypertext in the context of reading on the web: Language learners' experience,&quot; Current Issues in Education, vol. 6, no. 12, July, 2005. [Online serial]. Available: http://cie.ed.asu.edu/volume6/number12/. [Accessed Dec. 2, 2007]."
Loading

0 comments on commit ed39b74

Please sign in to comment.